diff --git a/rllib/utils/exploration/epsilon_greedy.py b/rllib/utils/exploration/epsilon_greedy.py
index 5ce114249..02dd0a20f 100644
--- a/rllib/utils/exploration/epsilon_greedy.py
+++ b/rllib/utils/exploration/epsilon_greedy.py
@@ -167,8 +167,7 @@ class EpsilonGreedy(Exploration):
                             exploit_action[j][i] = torch.tensor(
                                 random_action[j])
                 exploit_action = tree.unflatten_as(
-                    action_distribution.action_space_struct,
-                    exploit_action)
+                    action_distribution.action_space_struct, exploit_action)
 
                 return exploit_action, action_logp