Removed small bug