fixed the regression in the previous commit