--- a/agent/cli/src/test/java/com/redhat/thermostat/agent/cli/internal/ActivatorTest.java	Tue Jun 20 13:00:11 2017 +0200
+++ b/agent/cli/src/test/java/com/redhat/thermostat/agent/cli/internal/ActivatorTest.java	Tue Jun 20 14:44:16 2017 -0400
@@ -40,6 +40,8 @@
 import static org.junit.Assert.assertEquals;
 import static org.mockito.Mockito.mock;
 
+import com.redhat.thermostat.agent.dao.AgentInfoDAO;
+import com.redhat.thermostat.agent.dao.BackendInfoDAO;
 import org.junit.Test;
 
 import com.redhat.thermostat.common.ExitStatus;
@@ -58,6 +60,8 @@
         bundleContext.registerService(WriterID.class, writerID, null);
         bundleContext.registerService(ExitStatus.class, exitStatus, null);
         bundleContext.registerService(SSLConfiguration.class, mock(SSLConfiguration.class), null);
+        bundleContext.registerService(AgentInfoDAO.class, mock(AgentInfoDAO.class), null);
+        bundleContext.registerService(BackendInfoDAO.class, mock(BackendInfoDAO.class), null);
         
         Activator activator = new Activator();
 
@@ -65,14 +69,14 @@
         
         activator.start(bundleContext);
         
-        assertEquals(3, bundleContext.getServiceListeners().size());
+        assertEquals(5, bundleContext.getServiceListeners().size());
         
         assertCommandIsRegistered(bundleContext, "agent", AgentApplication.class);
 
         activator.stop(bundleContext);
 
         assertEquals(0, bundleContext.getServiceListeners().size());
-        assertEquals(3, bundleContext.getAllServices().size());
+        assertEquals(5, bundleContext.getAllServices().size());
     }
 }