Revert "removed unneeded code?"

This reverts commit 03dce795ad642c90ec1d83dec6c0974684ab6f9f.