diff --git a/tests/data/save_policy_to_safetensors/xarm_tdmpc/actions.safetensors b/tests/data/save_policy_to_safetensors/xarm_tdmpc/actions.safetensors
index 0339ca0e..49179928 100644
Binary files a/tests/data/save_policy_to_safetensors/xarm_tdmpc/actions.safetensors and b/tests/data/save_policy_to_safetensors/xarm_tdmpc/actions.safetensors differ