diff --git a/libs/RobotKit/tests/RobotController_test_hardware.cpp b/libs/RobotKit/tests/RobotController_test_hardware.cpp
index dca096f9a..dc5512630 100644
--- a/libs/RobotKit/tests/RobotController_test_hardware.cpp
+++ b/libs/RobotKit/tests/RobotController_test_hardware.cpp
@@ -39,6 +39,7 @@ TEST_F(RobotControllerTest, suspendHardwareForDeepSleep)
 	rc.registerDeepSleepEnabledComponents(components);
 
 	EXPECT_CALL(mock_deep_sleep_enabled, enableDeepSleep);
+	EXPECT_CALL(mbed_mock_gap, disconnect);
 	rc.suspendHardwareForDeepSleep();
 }
 
diff --git a/libs/RobotKit/tests/StateMachine_test.cpp b/libs/RobotKit/tests/StateMachine_test.cpp
index de9afa257..45db4abcb 100644
--- a/libs/RobotKit/tests/StateMachine_test.cpp
+++ b/libs/RobotKit/tests/StateMachine_test.cpp
@@ -272,6 +272,7 @@ TEST_F(StateMachineTest, stateDeepSleepingEventBleConnection)
 	sm.set_current_states(lksm::state::deep_sleeping);
 
 	EXPECT_CALL(mock_rc, wakeUp);
+	EXPECT_CALL(mock_rc, startConnectionBehavior);
 
 	sm.process_event(lksm::event::ble_connection {});