diff --git a/src/Command/Debug/ThemeKeysCommand.php b/src/Command/Debug/ThemeKeysCommand.php
index 69f8ab3..14b624c 100644
--- a/src/Command/Debug/ThemeKeysCommand.php
+++ b/src/Command/Debug/ThemeKeysCommand.php
@@ -11,13 +11,11 @@ use Symfony\Component\Console\Input\InputArgument;
 use Symfony\Component\Console\Input\InputInterface;
 use Symfony\Component\Console\Output\OutputInterface;
 use Symfony\Component\Yaml\Yaml;
-use Drupal\Console\Core\Command\Command;
-use Drupal\Console\Core\Command\Shared\CommandTrait;
+use Drupal\Console\Core\Command\ContainerAwareCommand;
 use Drupal\Core\Theme\Registry;
 
-class ThemeKeysCommand extends Command
+class ThemeKeysCommand extends ContainerAwareCommand
 {
-    use CommandTrait;
 
     /**
    * @var Registry