Revert "Small fix"

This reverts commit 26618882a5535228b3a8b72875723bfba7e02f42.