Forgot to uncomment a line in the previous commit. Should work now