diff --git a/src/Flagsmith.php b/src/Flagsmith.php
index 186a064..16058fa 100644
--- a/src/Flagsmith.php
+++ b/src/Flagsmith.php
@@ -256,9 +256,9 @@ public function withSkipCache(bool $skipCache): self
 
     /**
      * Get the environment model.
-     * @return EnvironmentModel
+     * @return EnvironmentModel|null
      */
-    public function getEnvironment(): EnvironmentModel
+    public function getEnvironment(): ?EnvironmentModel
     {
         return $this->environment;
     }