diff --git a/src/Di/Di.php b/src/Di/Di.php
index 1dbf969..3625d0c 100644
--- a/src/Di/Di.php
+++ b/src/Di/Di.php
@@ -125,9 +125,10 @@ public function attempt(string $name, $definition, bool $shared = false): Servic
     /**
      * Resolves the service based on its configuration
      *
-     * @param string $name
+     * @template T
+     * @param string|class-string<T> $name
      * @param mixed $parameters
-     * @return mixed
+     * @return mixed|T
      */
     public function get(string $name, $parameters = null): mixed
     {
diff --git a/src/Di/DiInterface.php b/src/Di/DiInterface.php
index 129a362..5450696 100644
--- a/src/Di/DiInterface.php
+++ b/src/Di/DiInterface.php
@@ -31,9 +31,10 @@ public function attempt(string $name, $definition, bool $shared = false): Servic
     /**
      * Resolves the service based on its configuration
      *
-     * @param string $name
+     * @template T
+     * @param string|class-string<T> $name
      * @param mixed $parameters
-     * @return mixed
+     * @return mixed|T
      */
     public function get(string $name, $parameters = null): mixed;