diff --git a/tensordict/nn/functional_modules.py b/tensordict/nn/functional_modules.py
index 011c0c753..7149d156a 100644
--- a/tensordict/nn/functional_modules.py
+++ b/tensordict/nn/functional_modules.py
@@ -200,7 +200,6 @@ def incompatible_error():
                 new_batch_size.insert(out_dim, batch_size)
                 new_names = list(batched_output.names)
                 new_names.insert(out_dim, None)
-                print(batched_output.names)
                 out = batched_output.apply(
                     lambda x, out_dim=out_dim: _remove_batch_dim(
                         x, vmap_level, batch_size, out_dim
@@ -208,7 +207,6 @@ def incompatible_error():
                     batch_size=new_batch_size,
                     names=new_names,
                 )
-                print(new_batch_size, out_dim, batch_size, flat_out_dims, out.names)
             flat_outputs.append(out)
         return tree_unflatten(flat_outputs, output_spec)