Fixed the small bug, blah