Removing unnecessary import