Fixing a bad import