Unit test fails due to cleanup