Cleanup of unit test