Cleanup testing code