change import statement in tests