Revert "Removed strange code"

This reverts commit 1d1b108e117faef060397bfb9cef9a3980c5020b.