Forgot to uncomment some code