Revert "Small fix"

This reverts commit 2103bbfaa27cb596803afe626550eefc2ad313f5.