Fix test cleanup for TestDXRun