Remove an unnessecery import