fix bug from previous commit for