Fixed the small bug :P