[#7] Fix test import.