small bug fix #2