diff --git a/douzero/env/env.py b/douzero/env/env.py
index 98fd9d2..c774018 100644
--- a/douzero/env/env.py
+++ b/douzero/env/env.py
@@ -195,7 +195,7 @@ class Env:
             }
             obs = None
         else:
-            obs = get_obs(self.infoset, self.use_general, self.use_legacy, self.lite_model)
+            obs = get_obs(self.infoset, self.use_general, self.use_legacy, self.lite_model, self.use_unified)
         return obs, reward, done, {}
 
     def _get_reward(self, pos):