Revert "Small fix"

This reverts commit 24da27540bae6e15b3470e4261ebb195f53d1b89.