Removed one small bug hidden by testing