Removing an unecessary import