fixed small bug in last commit