Fix small bug