Only include test code during testing

You know the drill. :-P