diff --git a/drivers/gnmi/src/main/java/org/onosproject/drivers/gnmi/AbstractGnmiHandlerBehaviour.java b/drivers/gnmi/src/main/java/org/onosproject/drivers/gnmi/AbstractGnmiHandlerBehaviour.java
index fda0bb5..e6d8b16 100644
--- a/drivers/gnmi/src/main/java/org/onosproject/drivers/gnmi/AbstractGnmiHandlerBehaviour.java
+++ b/drivers/gnmi/src/main/java/org/onosproject/drivers/gnmi/AbstractGnmiHandlerBehaviour.java
@@ -54,7 +54,7 @@
 
     protected boolean setupBehaviour() {
         deviceId = handler().data().deviceId();
-
+        deviceService = handler().get(DeviceService.class);
         controller = handler().get(GnmiController.class);
         client = controller.getClient(deviceId);