From 0fe5c35dda5fd66909f115611cb3eb6f680388a2 Mon Sep 17 00:00:00 2001
From: Dominik Roth <dominik.roth.dev@gmail.com>
Date: Wed, 26 Oct 2022 17:40:24 +0200
Subject: [PATCH] Small bug fix regarding aux_penalty

---
 columbus/env.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/columbus/env.py b/columbus/env.py
index bd96ce5..4bf9261 100644
--- a/columbus/env.py
+++ b/columbus/env.py
@@ -242,7 +242,7 @@ class ColumbusEnv(gym.Env):
                     or self.agent.pos[1] < 0.001 or self.agent.pos[1] > 0.999:
                 reward -= self.void_damage/self.fps
         self.score += reward  # aux_reward does not count towards the score
-        if self.aux_reward_max:
+        if self.aux_reward_max or self.aux_penalty_max:
             reward += self._get_aux_reward()
         self._steps += 1
         done = (self.die_on_zero and self.score <= 0) or (self.return_on_score != -