1 files changed, 4 insertions, 6 deletions
diff --git a/lib/tasks/gitlab/pages.rake b/lib/tasks/gitlab/pages.rake
index ee2931f0c4f..684d62d1367 100644
--- a/lib/tasks/gitlab/pages.rake
+++ b/lib/tasks/gitlab/pages.rake
@@ -61,26 +61,24 @@ namespace :gitlab do
     namespace :deployments do
       task migrate_to_object_storage: :gitlab_environment do
         logger = Logger.new(STDOUT)
-        logger.info('Starting transfer of pages deployments to remote storage')
 
-        helper = Gitlab::Pages::MigrationHelper.new(logger)
+        helper = Gitlab::LocalAndRemoteStorageMigration::PagesDeploymentMigrater.new(logger)
 
         begin
           helper.migrate_to_remote_storage
-        rescue => e
+        rescue StandardError => e
           logger.error(e.message)
         end
       end
 
       task migrate_to_local: :gitlab_environment do
         logger = Logger.new(STDOUT)
-        logger.info('Starting transfer of Pages deployments to local storage')
 
-        helper = Gitlab::Pages::MigrationHelper.new(logger)
+        helper = Gitlab::LocalAndRemoteStorageMigration::PagesDeploymentMigrater.new(logger)
 
         begin
           helper.migrate_to_local_storage
-        rescue => e
+        rescue StandardError => e
           logger.error(e.message)
         end
       end