Update tests to use get_import_data