Fixed a bug
Removed import