1 files changed, 3 insertions, 0 deletions
diff --git a/lib/gitlab/background_migration/user_mentions/models/design_management/design.rb b/lib/gitlab/background_migration/user_mentions/models/design_management/design.rb
index bdb90b5d2b9..d010d68600d 100644
--- a/lib/gitlab/background_migration/user_mentions/models/design_management/design.rb
+++ b/lib/gitlab/background_migration/user_mentions/models/design_management/design.rb
@@ -10,6 +10,9 @@ module Gitlab
             include EachBatch
             include Concerns::MentionableMigrationMethods
 
+            self.table_name = 'design_management_designs'
+            self.inheritance_column = :_type_disabled
+
             def self.user_mention_model
               Gitlab::BackgroundMigration::UserMentions::Models::DesignUserMention
             end