diff --git a/qa/qa/specs/features/ee/api/3_create/code_suggestions_spec.rb b/qa/qa/specs/features/ee/api/3_create/code_suggestions_spec.rb
index 7518fdb67b08f1b389bba3f092d0a5ddd3ceafcb..7c08f92167a1ecaff68fdfac76c1c91198dcfee7 100644
--- a/qa/qa/specs/features/ee/api/3_create/code_suggestions_spec.rb
+++ b/qa/qa/specs/features/ee/api/3_create/code_suggestions_spec.rb
@@ -116,6 +116,7 @@ def honk_horn(sound)
           context 'with context' do
             let(:prompt_data) { super().merge(context: context) }
 
+            it_behaves_like 'code suggestions API', 'https://gitlab.com/gitlab-org/gitlab/-/quality/test_cases/500005'
             it_behaves_like 'direct code completion', 'https://gitlab.com/gitlab-org/gitlab/-/quality/test_cases/491518'
           end
         end
@@ -131,6 +132,7 @@ def honk_horn(sound)
                 context 'with context' do
                   let(:prompt_data) { super().merge(context: context) }
 
+                  it_behaves_like 'code suggestions API', 'https://gitlab.com/gitlab-org/gitlab/-/quality/test_cases/500006'
                   it_behaves_like 'direct code completion', 'https://gitlab.com/gitlab-org/gitlab/-/quality/test_cases/491519'
                 end
               end