Forgoto to uncomment some code

that been commented out for testing