diff --git a/HelloWorldEnv.py b/HelloWorldEnv.py
index fc09bbf7ddb0aef471ab5d3d094d70b4d6c558ce..154e1354993671fcf90c720b7a7d82dc1a11d709 100644
--- a/HelloWorldEnv.py
+++ b/HelloWorldEnv.py
@@ -17,10 +17,10 @@ class HelloWorldEnv(PyEnvironment):
         self._client = airsim.MultirotorClient(ip=ip)
         super(HelloWorldEnv, self).__init__(handle_auto_reset)
 
-    def action_spec(self) -> NestedArraySpec:
+    def action_spec(self) -> BoundedArraySpec:
         return self._action_spec
     
-    def observation_spec(self) -> NestedArraySpec:
+    def observation_spec(self) -> ArraySpec:
         return self._observation_spec
     
     def _reset(self) -> TimeStep: