Minor fix in import_test