diff --git a/src/ServiceDefinitionBase.php b/src/ServiceDefinitionBase.php
index e29500c8251d15aebf270c5aa6213ee90c4be11d..959f7bcb1594cb0837ea0104fc09e7f4e2faf118 100644
--- a/src/ServiceDefinitionBase.php
+++ b/src/ServiceDefinitionBase.php
@@ -2,14 +2,16 @@
 
 namespace Drupal\services;
 
-use Drupal\Core\Plugin\ContextAwarePluginBase;
+use Drupal\Core\Plugin\ContextAwarePluginTrait;
+use Drupal\Core\Plugin\PluginBase;
 use Symfony\Component\HttpFoundation\Response;
 use Symfony\Component\Routing\Route;
 
 /**
  * Class \Drupal\services\ServiceDefinitionBase.
  */
-abstract class ServiceDefinitionBase extends ContextAwarePluginBase implements ServiceDefinitionInterface {
+abstract class ServiceDefinitionBase extends PluginBase implements ServiceDefinitionInterface {
+  use ContextAwarePluginTrait;
 
   /**
    * {@inheritdoc}