Updated small bug