Fixed a small bug (I'm tired)