Add cleanup section to test.