Cleanups in test case