diff --git a/spec/lib/gitlab/background_migration/prepare_untracked_uploads_spec.rb b/spec/lib/gitlab/background_migration/prepare_untracked_uploads_spec.rb
index 881d40b4404..35750d89c35 100644
--- a/spec/lib/gitlab/background_migration/prepare_untracked_uploads_spec.rb
+++ b/spec/lib/gitlab/background_migration/prepare_untracked_uploads_spec.rb
@@ -4,7 +4,7 @@ require 'spec_helper'
 describe Gitlab::BackgroundMigration::PrepareUntrackedUploads, :sidekiq, :migration, schema: 20180208183958 do
   include MigrationsHelpers::TrackUntrackedUploadsHelpers
 
-  let!(:untracked_files_for_uploads) { described_class::UntrackedFile }
+  let!(:untracked_files_for_uploads) { table(:untracked_files_for_uploads) }
   let!(:appearances) { table(:appearances) }
   let!(:namespaces) { table(:namespaces) }
   let!(:projects) { table(:projects) }