small bug fix on my previous commit