Fix test for import.