finish unit tests for new import_stage