diff --git a/tests/data/save_policy_to_safetensors/xarm_tdmpc/output_dict.safetensors b/tests/data/save_policy_to_safetensors/xarm_tdmpc/output_dict.safetensors
index 2321f31c..ef49ce97 100644
Binary files a/tests/data/save_policy_to_safetensors/xarm_tdmpc/output_dict.safetensors and b/tests/data/save_policy_to_safetensors/xarm_tdmpc/output_dict.safetensors differ