diff --git a/scheduled_updates.services.yml b/scheduled_updates.services.yml
index ee44b36..a08dba1 100644
--- a/scheduled_updates.services.yml
+++ b/scheduled_updates.services.yml
@@ -21,4 +21,9 @@ services:
     arguments: ['@entity.manager']
     tags:
       - { name: route_enhancer }
+  scheduled_updates.run_updates:
+    class: Drupal\scheduled_updates\Command\RunUpdatesCommand
+    arguments: ['@scheduled_updates.update_runner']
+    tags:
+      - { name: drupal.command }
 #rename scheduled_updates.update_utils service or class
diff --git a/src/Command/RunUpdatesCommand.php b/src/Command/RunUpdatesCommand.php
index aedf25c..628c30b 100644
--- a/src/Command/RunUpdatesCommand.php
+++ b/src/Command/RunUpdatesCommand.php
@@ -7,19 +7,36 @@
 
 namespace Drupal\scheduled_updates\Command;
 
-use Drupal\Console\Command\ContainerAwareCommand;
-use Drupal\Console\Style\DrupalStyle;
+use Drupal\Console\Core\Command\Shared\ContainerAwareCommandTrait;
+use Drupal\Console\Core\Style\DrupalStyle;
 use Drupal\scheduled_updates\UpdateRunnerUtils;
 use Symfony\Component\Console\Input\InputArgument;
 use Symfony\Component\Console\Input\InputInterface;
 use Symfony\Component\Console\Output\OutputInterface;
+use Symfony\Component\Console\Command\Command;
 
 /**
  * Class RunUpdatesCommand.
  *
  * @package Drupal\scheduled_updates
  */
-class RunUpdatesCommand extends ContainerAwareCommand {
+class RunUpdatesCommand extends Command {
+
+  /** @var UpdateRunnerUtils $runnerUtils */
+  protected $runnerUtils;
+
+  /**
+   * RunUpdatesCommand constructor.
+   * @param UpdateRunnerUtils $runnerUtils
+   */
+  public function __construct(
+      UpdateRunnerUtils $runnerUtils
+  ) {
+    $this->runnerUtils = $runnerUtils;
+    parent::__construct();
+  }
+
+  use ContainerAwareCommandTrait;
   /**
    * {@inheritdoc}
    */
@@ -35,8 +52,8 @@ class RunUpdatesCommand extends ContainerAwareCommand {
    */
   protected function execute(InputInterface $input, OutputInterface $output) {
     $io = new DrupalStyle($input, $output);
-    /** @var UpdateRunnerUtils $runnerUtils */
-    if ($runnerUtils = $this->hasGetService('scheduled_updates.update_runner')) {
+
+    if ($this->runnerUtils) {
       $update_types = $input->getArgument('update_types');
       if ($update_types) {
         $update_types = explode(',', $update_types);
@@ -44,7 +61,7 @@ class RunUpdatesCommand extends ContainerAwareCommand {
       else {
         $update_types = [];
       }
-      $runnerUtils->runAllUpdates($update_types);
+      $this->runnerUtils->runAllUpdates($update_types);
       $io->info('Updates run!');
     }
     else {