diff --git a/examples/animation/bayes_update.py b/examples/animation/bayes_update.py
index a05930562512..4276e62b864f 100644
--- a/examples/animation/bayes_update.py
+++ b/examples/animation/bayes_update.py
@@ -40,8 +40,7 @@ def __call__(self, i):
         self.line.set_data(self.x, y)
         return self.line,
 
-fig = plt.figure()
-ax = fig.add_subplot(1, 1, 1)
+fig, ax = plt.subplots()
 ud = UpdateDist(ax, prob=0.7)
 anim = FuncAnimation(fig, ud, frames=np.arange(100), init_func=ud.init,
                      interval=100, blit=True)