Revert "Small fix"

This reverts commit 5eb7440513af6d08a51d5e11006c20eb67a032ac.