Fixing extra import