fix test file missing import