fix import error in tests