Fix the import --edit function