Removing an unnecessary import