Fix unneccesary import