diff --git a/Classes/Traits/DependencyInjectionTrait.php b/Classes/Traits/DependencyInjectionTrait.php
index d2ccc13..039f1c3 100644
--- a/Classes/Traits/DependencyInjectionTrait.php
+++ b/Classes/Traits/DependencyInjectionTrait.php
@@ -111,6 +111,11 @@ trait DependencyInjectionTrait
             get_class($this)
         );
         foreach ($classInfo->getInjectMethods() as $method => $className) {
+            if (version_compare(TYPO3_branch, '10.0', '>=')) {
+                $className = $className
+                    ->getFirstParameter()
+                    ->getDependency();
+            }
             $class = $this->objectManager->get(
                 $className
             );