fix imports on some tests
2 files changed