Only include test code during testing