diff --git a/src/test/java/net/floodlightcontroller/core/internal/OFChannelHandlerTest.java b/src/test/java/net/floodlightcontroller/core/internal/OFChannelHandlerTest.java
index ca13841030a5b2c2d39d7df29ab53cbaec9a4c58..56b5f2e3c44175294a87d91a1197bd80cfb388df 100644
--- a/src/test/java/net/floodlightcontroller/core/internal/OFChannelHandlerTest.java
+++ b/src/test/java/net/floodlightcontroller/core/internal/OFChannelHandlerTest.java
@@ -515,6 +515,10 @@ public class OFChannelHandlerTest {
         expectLastCall().once();
         sw.setCoreFlowPriority(CORE_PRIORITY);
         expectLastCall().once();
+        sw.startDriverHandshake();
+        expectLastCall().once();
+        sw.isDriverHandshakeComplete();
+        expectLastCall().andReturn(true).once();
 
         if (cfg.isPresent)
             sw.setAttribute(IOFSwitch.SWITCH_IS_CORE_SWITCH, cfg.isCoreSwitch);