Cleaning up testing code.