diff --git a/examples/ClientTests/scripts/Check_Air.py b/examples/ClientTests/scripts/Check_Air.py
index 1005ae258ecb15b981aedbe41a6e989e7dd299df..cfcb7076e21483ba3b873ee6f0b703b6a577507d 100644
--- a/examples/ClientTests/scripts/Check_Air.py
+++ b/examples/ClientTests/scripts/Check_Air.py
@@ -42,8 +42,8 @@ def run():
   # Configure UI
   geant4 = DDG4.Geant4(kernel, tracker='Geant4TrackerCombineAction')
   if batch:
-    ui = geant4.setupCshUI(ui=None, vis=None)
-    kernel.UI = 'UI'
+    ui = geant4.setupCshUI(typ=None, ui=None, vis=None)
+    kernel.UI = ui.name
   else:
     ui = geant4.setupCshUI(vis=vis)
   kernel.loadGeometry(geo)