fixed small bug in implementation