fixing incorrect test import