Removing a couple of unnecessary lines