A simple change to make testing conditional.

I have not tested it much but it should just work.