diff --git a/src/Tempest/Container/src/AutoDiscovery.php b/src/Tempest/Container/src/AutoDiscovery.php
index ad87bf8ad..b90a74f61 100644
--- a/src/Tempest/Container/src/AutoDiscovery.php
+++ b/src/Tempest/Container/src/AutoDiscovery.php
@@ -8,6 +8,9 @@
 use Tempest\Core\HandlesDiscoveryCache;
 use Tempest\Reflection\ClassReflector;
 
+/**
+ * @property GenericContainer $container
+ */
 final class AutoDiscovery implements Discovery
 {
     use HandlesDiscoveryCache;
diff --git a/src/Tempest/Container/src/GenericContainer.php b/src/Tempest/Container/src/GenericContainer.php
index 9c7c18ac5..da59ae70a 100644
--- a/src/Tempest/Container/src/GenericContainer.php
+++ b/src/Tempest/Container/src/GenericContainer.php
@@ -39,6 +39,8 @@ public function __construct(
     public function setDefinitions(array $definitions): self
     {
         $this->definitions = new ArrayIterator($definitions);
+
+        return $this;
     }
 
     public function setInitializers(array $initializers): self