Uncomment necessary code

I commented these two lines when I was testing and neglected to uncomment them prior to my last pull request. Sorry about that.