Fixing problem after import