Fixed test import