minor bug from previous commit