rectify small bug