diff --git a/torchy_baselines/common/base_class.py b/torchy_baselines/common/base_class.py
index f517c58..4c7a764 100644
--- a/torchy_baselines/common/base_class.py
+++ b/torchy_baselines/common/base_class.py
@@ -596,6 +596,7 @@ class BaseRLModel(object):
                 total_episodes += 1
                 episode_rewards.append(episode_reward)
                 total_timesteps.append(episode_timesteps)
+                # TODO: reset SDE matrix at the end of the episode?
                 if action_noise is not None:
                     action_noise.reset()