Removing unncesary import