diff --git a/libcontainer/integration/checkpoint_test.go b/libcontainer/integration/checkpoint_test.go
index 2d13469da..311e33c99 100644
--- a/libcontainer/integration/checkpoint_test.go
+++ b/libcontainer/integration/checkpoint_test.go
@@ -196,12 +196,7 @@ func testCheckpoint(t *testing.T, userns bool) {
 	ok(t, err)
 
 	restoreStdinW.Close()
-	s, err := restoreProcessConfig.Wait()
-	ok(t, err)
-
-	if !s.Success() {
-		t.Fatal(s.String(), pid)
-	}
+	waitProcess(restoreProcessConfig, t)
 
 	output := restoreStdout.String()
 	if !strings.Contains(output, "Hello!") {