Corrected small bug