fix import in tests