Forgot to uncomment some lines that were used for testing