Fix small bug.