update test codes
diff --git a/src/test/java/net/onrc/onos/util/GraphDBOperationTest.java b/src/test/java/net/onrc/onos/util/GraphDBOperationTest.java
index 04ef815..01f244c 100644
--- a/src/test/java/net/onrc/onos/util/GraphDBOperationTest.java
+++ b/src/test/java/net/onrc/onos/util/GraphDBOperationTest.java
@@ -86,8 +86,7 @@
 		switches = op.getAllSwitches();
 		assertFalse(switches.iterator().hasNext());
 
-		ISwitchObject sw = op.newSwitch();
-		sw.setDPID("123");
+		ISwitchObject sw = op.newSwitch("123");
 		sw.setState(SwitchState.ACTIVE.toString());
 		conn.endTx(Transaction.COMMIT);
 
@@ -110,8 +109,7 @@
 		switches = op.getAllSwitches();
 		assertFalse(switches.iterator().hasNext());
 		
-		ISwitchObject sw = op.newSwitch();
-		sw.setDPID("123");
+		ISwitchObject sw = op.newSwitch("123");
 		sw.setState(SwitchState.ACTIVE.toString());
 		conn.endTx(Transaction.COMMIT);