diff --git a/torchrl/collectors/collectors.py b/torchrl/collectors/collectors.py
index 1a9e445316d..4a83258278d 100644
--- a/torchrl/collectors/collectors.py
+++ b/torchrl/collectors/collectors.py
@@ -189,7 +189,7 @@ def _get_policy_and_device(
                     "The collector will trust that the devices match. To suppress this "
                     "warning, set `trust_policy=True` when building the collector."
                 )
-                return policy, None
+            return policy, None
 
         def map_weight(
             weight,