diff --git a/src/cmsstyle/cmsstyle.py b/src/cmsstyle/cmsstyle.py
index 4e78fcf..733f55b 100644
--- a/src/cmsstyle/cmsstyle.py
+++ b/src/cmsstyle/cmsstyle.py
@@ -6,7 +6,6 @@
 import ROOT as rt
 from array import array
 
-rt.gROOT.SetBatch(rt.kTRUE)
 
 cms_lumi = "Run 2, 138 fb^{#minus1}"
 cms_energy = "13 TeV"
diff --git a/tests/example.py b/tests/example.py
index b986a13..4257d1d 100644
--- a/tests/example.py
+++ b/tests/example.py
@@ -2,6 +2,7 @@
 import cmsstyle as CMS
 
 CMS.SetExtraText("Simulation Preliminary")
+ROOT.gROOT.SetBatch(ROOT.kTRUE)
 
 
 class Plotter:
diff --git a/tests/example_palette.py b/tests/example_palette.py
index b35e078..9f5d6dc 100644
--- a/tests/example_palette.py
+++ b/tests/example_palette.py
@@ -2,6 +2,7 @@
 import cmsstyle as CMS
 
 CMS.SetExtraText("Simulation")
+ROOT.gROOT.SetBatch(ROOT.kTRUE)
 
 class Plotter:
     def __init__(self):