Adding better unit test for import/export