fix another other bug from last commit