fix test import for tests