diff --git a/olaflow/processing/diff.py b/olaflow/processing/diff.py
index 6f41763..c8788b1 100644
--- a/olaflow/processing/diff.py
+++ b/olaflow/processing/diff.py
@@ -57,8 +57,8 @@ def get_pickle(out):
 
 models = list(map(get_pickle, args.output))
 
+fig, ax = plt.subplots(len(models), figsize=(6, 1.5 * len(models)), dpi=100, constrained_layout=True)
 if args.timestep is None:
-    fig, ax = plt.subplots(len(models), constrained_layout=True)
     for i, (_ax, _model) in enumerate(zip(ax, models)):
         _ax.contour(
             _model.t,
@@ -77,7 +77,6 @@ if args.timestep is None:
         )
     )
 else:
-    fig, ax = plt.subplots(len(models), constrained_layout=True)
     for i, (_ax, _model) in enumerate(zip(ax, models)):
         _ax.tricontour(
             _model.x,