fix error from previous commit