Spiking some code for the test class generator