Updated import_test slightly