fix test import