1 files changed, 5 insertions, 5 deletions
diff --git a/same/src/main/java/com/orbekk/same/TestConnectionManager.java b/same/src/main/java/com/orbekk/same/TestConnectionManager.java
index 6f5e5d8..93b2e6a 100644
--- a/same/src/main/java/com/orbekk/same/TestConnectionManager.java
+++ b/same/src/main/java/com/orbekk/same/TestConnectionManager.java
@@ -4,7 +4,7 @@ import java.util.Map;
 import java.util.HashMap;
 
 import com.orbekk.paxos.PaxosService;
-import com.orbekk.same.discovery.DirectoryService;
+import com.orbekk.same.Services.Directory;
 
 /**
  * This class is used in test.
@@ -16,8 +16,8 @@ public class TestConnectionManager implements ConnectionManager {
             new HashMap<String, MasterService>();
     public Map<String, PaxosService> paxosMap =
             new HashMap<String, PaxosService>();
-    public Map<String, DirectoryService> directoryMap =
-            new HashMap<String, DirectoryService>();
+    public Map<String, Services.Directory> directoryMap =
+            new HashMap<String, Services.Directory>();
 
     public TestConnectionManager() {
     }
@@ -38,7 +38,7 @@ public class TestConnectionManager implements ConnectionManager {
     }
 
     @Override
-    public DirectoryService getDirectory(String url) {
-        return directoryMap.get(url);
+    public Directory getDirectory(String location) {
+        return directoryMap.get(location);
     }
 }