Revert "Small fix"

This reverts commit 5c2e0ea1f511bc2c8a14c5bfe40d96e6df9b6ff0.