Fix for unit test of import