fixing test_restart failures