Fix for bad disconnect during test run.