diff --git a/test/runtime/assistant_execution_target_test.dart b/test/runtime/assistant_execution_target_test.dart
index 8ee1d4f5..f022c928 100644
--- a/test/runtime/assistant_execution_target_test.dart
+++ b/test/runtime/assistant_execution_target_test.dart
@@ -97,14 +97,8 @@ void main() {
           'session-1',
         );
         expect(
-          controller.assistantExecutionTargetForSession('session-1').isGateway,
-          isTrue,
-        );
-        expect(
-          assistantExecutionTargetFromExecutionMode(
-            record.executionBinding.executionMode,
-          ),
-          AssistantExecutionTarget.gateway,
+          record.executionBinding.executionMode,
+          ThreadExecutionMode.gateway,
         );
         expect(
           controller.assistantProviderForSession('session-1'),