diff --git a/ee/spec/migrations/20250214085204_fill_ai_duo_chat_events_organization_id_spec.rb b/ee/spec/migrations/20250214085204_fill_ai_duo_chat_events_organization_id_spec.rb
index 46d99764c6fca162a4f5c8f5256989f5cfbe6b05..7858c52a400532b176175879bd18cac96b5997fa 100644
--- a/ee/spec/migrations/20250214085204_fill_ai_duo_chat_events_organization_id_spec.rb
+++ b/ee/spec/migrations/20250214085204_fill_ai_duo_chat_events_organization_id_spec.rb
@@ -13,7 +13,7 @@
       organization_id: organization.id)
   end
 
-  let(:ai_duo_chat_events) { table(:ai_duo_chat_events) }
+  let(:ai_duo_chat_events) { partitioned_table(:ai_duo_chat_events) }
 
   let!(:event1) do
     ai_duo_chat_events.create!(