fixed small bug in output