oops, fix small bug from last commit