fix another bug from last commit