fixed small bug with comments