diff --git a/runtime_test.go b/runtime_test.go
index f9a209008c..5fa6c46bfe 100644
--- a/runtime_test.go
+++ b/runtime_test.go
@@ -457,7 +457,7 @@ func TestRestore(t *testing.T) {
 
 	// Here are are simulating a docker restart - that is, reloading all containers
 	// from scratch
-	runtime2, err := NewRuntimeFromDirectory(runtime1.root, devmapper.NewDeviceSetDM(runtime1.root), false)
+	runtime2, err := NewRuntimeFromDirectory(runtime1.root, runtime1.deviceSet, false)
 	if err != nil {
 		t.Fatal(err)
 	}
diff --git a/utils_test.go b/utils_test.go
index bc4dd65a3c..f5bdac3271 100644
--- a/utils_test.go
+++ b/utils_test.go
@@ -2,7 +2,7 @@ package docker
 
 import (
 	"github.com/dotcloud/docker/utils"
-	"github.com/dotcloud/docker/devmapper"
+	"path/filepath"
 	"io"
 	"io/ioutil"
 	"os"
@@ -43,7 +43,7 @@ func newTestRuntime() (*Runtime, error) {
 		return nil, err
 	}
 
-	runtime, err := NewRuntimeFromDirectory(root, devmapper.NewDeviceSetDM(root), false)
+	runtime, err := NewRuntimeFromDirectory(root, NewDeviceSetWrapper (globalRuntime.deviceSet, filepath.Base(root)), false)
 	if err != nil {
 		return nil, err
 	}