diff --git a/libs/BehaviorKit/tests/BehaviorKit_test.cpp b/libs/BehaviorKit/tests/BehaviorKit_test.cpp
index 30acf637ae..b6a0303b6c 100644
--- a/libs/BehaviorKit/tests/BehaviorKit_test.cpp
+++ b/libs/BehaviorKit/tests/BehaviorKit_test.cpp
@@ -76,8 +76,11 @@ TEST_F(BehaviorKitTest, startBehaviorIDNotRegistered)
 	});
 	behaviorkit.registerBehaviors(behaviors);
 
-	EXPECT_CALL(mock_behavior_a, id);
-	EXPECT_CALL(mock_behavior_b, id);
+	auto behavior_a_id = BehaviorID {1};
+	auto behavior_b_id = BehaviorID {2};
+
+	EXPECT_CALL(mock_behavior_a, id).WillOnce(Return(behavior_a_id));
+	EXPECT_CALL(mock_behavior_b, id).WillOnce(Return(behavior_b_id));
 
 	behaviorkit.start(3);
 }