[FIX][TEST] fix test import