Hotfix, forgot to uncomment a piece of code