fixing a small bug in last commit