fixing small bug in last commit