fix accidental bug from last commit