Fixing import name bug