small bug fix in latest commit