diff --git a/src/Container/Traits/HasExtensions.php b/src/Container/Traits/HasExtensions.php
index 475691b..811f3e5 100644
--- a/src/Container/Traits/HasExtensions.php
+++ b/src/Container/Traits/HasExtensions.php
@@ -64,7 +64,7 @@ protected function resolve(\Xefi\Faker\Extensions\Extension|string $extension):
      *
      * @param \Xefi\Faker\Extensions\Extension $extension
      *
-     * @return \Xefi\Faker\Extensions\Extension
+     * @return \Xefi\Faker\Container\Container
      */
     protected function addExtension(Extension $extension): Container
     {
@@ -96,7 +96,7 @@ protected function addExtension(Extension $extension): Container
      *
      * @param \Xefi\Faker\Extensions\Extension $extension
      *
-     * @return \Xefi\Faker\Extensions\Extension
+     * @return \Xefi\Faker\Container\Container
      */
     protected function addLocaleExtension(Extension $extension): Container
     {