Removed an unnecessary import