@@ -256,6 +256,10 @@ func TestRmi(t *testing.T) {

 		t.Fatal(err)

 	}

+	if _, err := srv.ContainerWait(containerID); err != nil {

+		t.Fatal(err)

+	}

+

 	imageID, err := srv.ContainerCommit(containerID, "test", "", "", "", nil)

 	if err != nil {

 		t.Fatal(err)

@@ -277,6 +281,10 @@ func TestRmi(t *testing.T) {

 		t.Fatal(err)

 	}

+	if _, err := srv.ContainerWait(containerID); err != nil {

+		t.Fatal(err)

+	}

+

 	_, err = srv.ContainerCommit(containerID, "test", "", "", "", nil)

 	if err != nil {

 		t.Fatal(err)