Removed a small bug