diff --git a/db/migrate/20241025135522_alter_ai_ids_to_bigint.rb b/db/migrate/20241025135522_alter_ai_ids_to_bigint.rb
new file mode 100644
index 000000000..b94d0b4d3
--- /dev/null
+++ b/db/migrate/20241025135522_alter_ai_ids_to_bigint.rb
@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+class AlterAiIdsToBigint < ActiveRecord::Migration[7.1]
+  def up
+    change_column :ai_document_fragment_embeddings, :rag_document_fragment_id, :bigint
+    change_column :classification_results, :target_id, :bigint
+    change_column :rag_document_fragments, :target_id, :bigint
+  end
+
+  def down
+    raise ActiveRecord::IrreversibleMigration
+  end
+end
diff --git a/spec/plugin_helper.rb b/spec/plugin_helper.rb
index 0a2038342..2c7d9d7aa 100644
--- a/spec/plugin_helper.rb
+++ b/spec/plugin_helper.rb
@@ -17,14 +17,4 @@ def assign_fake_provider_to(setting_name)
   end
 end
 
-RSpec.configure do |config|
-  config.include DiscourseAi::ChatBotHelper
-
-  config.before(:suite) do
-    if defined?(migrate_column_to_bigint)
-      migrate_column_to_bigint(RagDocumentFragment, :target_id)
-      migrate_column_to_bigint("ai_document_fragment_embeddings", "rag_document_fragment_id")
-      migrate_column_to_bigint(ClassificationResult, :target_id)
-    end
-  end
-end
+RSpec.configure { |config| config.include DiscourseAi::ChatBotHelper }