Revert "Small Fix"

This reverts commit 2a34fbf39caec6b1a480f096022ec136d32b4476.