fix bug from previous commit