@@ -37,7 +37,7 @@ func TestInterruptedRegister(t *testing.T) {

 		Comment: "testing",

 		Created: time.Now(),

 	}

-	go graph.Register(badArchive, image)

+	go graph.Register(badArchive, false, image)

 	time.Sleep(200 * time.Millisecond)

 	w.CloseWithError(errors.New("But I'm not a tarball!")) // (Nobody's perfect, darling)

 	if _, err := graph.Get(image.Id); err == nil {

@@ -48,7 +48,7 @@ func TestInterruptedRegister(t *testing.T) {

 	if err != nil {

 		t.Fatal(err)

 	}

-	if err := graph.Register(goodArchive, image); err != nil {

+	if err := graph.Register(goodArchive, false, image); err != nil {

 		t.Fatal(err)

 	}

 }

@@ -94,7 +94,7 @@ func TestRegister(t *testing.T) {

 		Comment: "testing",

 		Created: time.Now(),

 	}

-	err = graph.Register(archive, image)

+	err = graph.Register(archive, false, image)

 	if err != nil {

 		t.Fatal(err)

 	}

@@ -212,7 +212,7 @@ func TestDelete(t *testing.T) {

 	assertNImages(graph, t, 1)

 	// Test delete twice (pull -> rm -> pull -> rm)

-	if err := graph.Register(archive, img1); err != nil {

+	if err := graph.Register(archive, false, img1); err != nil {

 		t.Fatal(err)

 	}

 	if err := graph.Delete(img1.Id); err != nil {