diff --git a/mani_skill/agents/registration.py b/mani_skill/agents/registration.py
index 203909d1b..5c9c82e61 100644
--- a/mani_skill/agents/registration.py
+++ b/mani_skill/agents/registration.py
@@ -32,7 +32,7 @@ def _register_agent(agent_cls: type[BaseAgent]):
                 logger.warn(
                     f"Agent {agent_cls.uid} is already registered. Skip registration."
                 )
-            return agent_cls
+                return agent_cls
 
         REGISTERED_AGENTS[agent_cls.uid] = AgentSpec(
             agent_cls=agent_cls, asset_download_ids=asset_download_ids