small bug fix that i created in last commit