minor update in comment
small bug fix