diff --git a/src/mouseinfo/__init__.py b/src/mouseinfo/__init__.py
index 08198cd..c2959fa 100644
--- a/src/mouseinfo/__init__.py
+++ b/src/mouseinfo/__init__.py
@@ -220,9 +220,11 @@ def _macGetPixel(x, y):
         else:
             raise
 
-    _display = Display(os.environ['DISPLAY'])
+    _display = Display(os.environ['DISPLAY']) if 'DISPLAY' in os.environ else None
 
     def _linuxPosition():
+        if _display is None:
+            return
         coord = _display.screen().root.query_pointer()._data
         return coord["root_x"], coord["root_y"]
     position = _linuxPosition
@@ -251,6 +253,8 @@ def _linuxScreenshot(filename=None):
     screenshot = _linuxScreenshot
 
     def _linuxSize():
+        if _display is None:
+            return
         return _display.screen().width_in_pixels, _display.screen().height_in_pixels
     size = _linuxSize