diff --git a/tutorials/dataframe/df104_HiggsToTwoPhotons.py b/tutorials/dataframe/df104_HiggsToTwoPhotons.py
index b5ad9e3c9dc74033ce28f445340c58515f283d8f..a53dec2fd473772ca161700bc79225b08950d428 100644
--- a/tutorials/dataframe/df104_HiggsToTwoPhotons.py
+++ b/tutorials/dataframe/df104_HiggsToTwoPhotons.py
@@ -76,7 +76,12 @@ for p in processes:
             ROOT.ROOT.RDF.TH1DModel(p, "Diphoton invariant mass; m_{#gamma#gamma} [GeV];Events", 30, 105, 160),
             "m_yy", "weight")
 
-# Run the event loop and create the plot
+# Run the event loop
+ggh = hists["ggH"].GetValue()
+vbf = hists["VBF"].GetValue()
+data = hists["data"].GetValue()
+
+# Create the plot
 
 # Set styles
 ROOT.gROOT.SetStyle("ATLAS")
@@ -98,11 +103,6 @@ lower_pad.SetBottomMargin(0.3)
 upper_pad.Draw()
 lower_pad.Draw()
 
-# Run the event loop
-ggh = hists["ggH"].GetValue()
-vbf = hists["VBF"].GetValue()
-data = hists["data"].GetValue()
-
 # Fit signal + background model to data
 upper_pad.cd()
 fit = ROOT.TF1("fit", "([0]+[1]*x+[2]*x^2+[3]*x^3)+[4]*exp(-0.5*((x-[5])/[6])^2)", 105, 160)
@@ -209,6 +209,5 @@ text.DrawLatex(0.18 + 0.13, 0.84, "Open Data")
 text.SetTextSize(0.04)
 text.DrawLatex(0.18, 0.78, "#sqrt{s} = 13 TeV, 10 fb^{-1}");
 
-# Draw and save the plot
-c.Draw()
+# Save the plot
 c.SaveAs("HiggsToTwoPhotons.pdf");