Small fix
Revert "Small fix"

This reverts commit 8c617b0102bf07194a68c94d7053ae44a5e62c79.