diff --git a/kalman_filter/display/display_2d.py b/kalman_filter/display/display_2d.py
index b2c9209..e8a8d52 100644
--- a/kalman_filter/display/display_2d.py
+++ b/kalman_filter/display/display_2d.py
@@ -87,6 +87,7 @@ class Displayer:
 
         for t in range(1, n_steps):
             callback(self, t)
+            self._ax.legend()
 
             if t in frames:
                 self._fig.savefig(prefix + str(t) + "." + ext)