diff --git a/tuhi-gui.in b/tuhi-gui.in
index b5786f3..b64edf2 100755
--- a/tuhi-gui.in
+++ b/tuhi-gui.in
@@ -39,10 +39,14 @@ def gtk_style():
     }
     """
 
+    screen = Gdk.Screen.get_default()
+    if screen is None:
+        print('Error: Unable to connect to screen. Make sure DISPLAY or WAYLAND_DISPLAY are set', file=sys.stderr)
+        sys.exit(1)
     style_provider = Gtk.CssProvider()
     style_provider.load_from_data(css)
     Gtk.StyleContext.add_provider_for_screen(
-            Gdk.Screen.get_default(),
+            screen,
             style_provider,
             Gtk.STYLE_PROVIDER_PRIORITY_APPLICATION
             )