small bug fix in previous commit