diff --git a/src/test/java/net/onrc/onos/core/intent/runtime/PathCalcRuntimeModuleTest.java b/src/test/java/net/onrc/onos/core/intent/runtime/PathCalcRuntimeModuleTest.java
index 93f01ca..421d2fa 100644
--- a/src/test/java/net/onrc/onos/core/intent/runtime/PathCalcRuntimeModuleTest.java
+++ b/src/test/java/net/onrc/onos/core/intent/runtime/PathCalcRuntimeModuleTest.java
@@ -85,8 +85,7 @@
         persistIntent = PowerMock.createMock(PersistIntent.class);
 
         PowerMock.expectNew(PersistIntent.class,
-                anyObject(IControllerRegistryService.class),
-                anyObject(INetworkGraphService.class)).andReturn(persistIntent);
+                anyObject(IControllerRegistryService.class)).andReturn(persistIntent);
 
         expect(modContext.getServiceImpl(IDatagridService.class))
                 .andReturn(datagridService).once();
diff --git a/src/test/java/net/onrc/onos/core/intent/runtime/UseCaseTest.java b/src/test/java/net/onrc/onos/core/intent/runtime/UseCaseTest.java
index fbb7313..0e58739 100755
--- a/src/test/java/net/onrc/onos/core/intent/runtime/UseCaseTest.java
+++ b/src/test/java/net/onrc/onos/core/intent/runtime/UseCaseTest.java
@@ -76,8 +76,7 @@
         persistIntent = PowerMock.createMock(PersistIntent.class);
 
         PowerMock.expectNew(PersistIntent.class,
-                anyObject(IControllerRegistryService.class),
-                anyObject(INetworkGraphService.class)).andReturn(persistIntent);
+                anyObject(IControllerRegistryService.class)).andReturn(persistIntent);
 
         expect(modContext.getServiceImpl(IDatagridService.class))
                 .andReturn(datagridService).once();