rectified a small bug