diff --git a/runtime_test.go b/runtime_test.go
index f76df953be..355d222e06 100644
--- a/runtime_test.go
+++ b/runtime_test.go
@@ -358,4 +358,5 @@ func TestRestore(t *testing.T) {
 	if err := container3.Run(); err != nil {
 		t.Fatal(err)
 	}
+	container2.State.Running = false
 }