Revert "small fix"

This reverts commit a8efa1dab36fac63bcd13b04f565bd48f8491c52.