fix import for tests