Revert "Revert "small fix""

This reverts commit 7966503091d83eaae9a45f2a6f370e32bdc679d2.