diff --git a/graph_test.go b/graph_test.go
index d89e6efe9d..ecb5ffb34e 100644
--- a/graph_test.go
+++ b/graph_test.go
@@ -152,7 +152,7 @@ func TestMount(t *testing.T) {
 	}
 	// FIXME: test for mount contents
 	defer func() {
-		if err := Unmount(rootfs); err != nil {
+		if err := image.Unmount(runtime, rootfs, "testing"); err != nil {
 			t.Error(err)
 		}
 	}()