Optimized code
diff --git a/src/main/java/net/onrc/onos/graph/DBOperation.java b/src/main/java/net/onrc/onos/graph/DBOperation.java
index f9b0e9f..1f77728 100644
--- a/src/main/java/net/onrc/onos/graph/DBOperation.java
+++ b/src/main/java/net/onrc/onos/graph/DBOperation.java
@@ -116,9 +116,8 @@
 
     }
 
-    @Override
-    public IDeviceObject newDevice() {
-        IDeviceObject obj = (IDeviceObject) conn.getFramedGraph().addVertex(null, IDeviceObject.class);
+    public IDeviceObject newDevice(final FramedGraph fg) {
+        IDeviceObject obj = (IDeviceObject) fg.addVertex(null, IDeviceObject.class);
         if (obj != null) {
             obj.setType("device");
         }