diff --git a/dizoo/petting_zoo/config/ptz_simple_spread_madqn_config.py b/dizoo/petting_zoo/config/ptz_simple_spread_madqn_config.py
index 1433bc0e17..b7db69abbe 100644
--- a/dizoo/petting_zoo/config/ptz_simple_spread_madqn_config.py
+++ b/dizoo/petting_zoo/config/ptz_simple_spread_madqn_config.py
@@ -27,6 +27,7 @@
             obs_shape=2 + 2 + n_landmark * 2 + (n_agent - 1) * 2 + (n_agent - 1) * 2,
             global_obs_shape=2 + 2 + n_landmark * 2 + (n_agent - 1) * 2 + (n_agent - 1) * 2 + n_agent * (2 + 2) +
             n_landmark * 2 + n_agent * (n_agent - 1) * 2,
+            agent_num=n_agent,
             action_shape=5,
             global_cooperation=True,
             hidden_size_list=[256, 256],