diff --git a/examples/urgslam.py b/examples/urgslam.py
index 2e2ba09..be8236c 100755
--- a/examples/urgslam.py
+++ b/examples/urgslam.py
@@ -59,8 +59,7 @@ if __name__ == '__main__':
 
         display.setPose(x, y, theta)
 
-        # Exit on ESCape
-        key = display.refresh()
-        if key != None and (key&0x1A):
+        # Exit on window close
+        if not display.refresh():
             exit(0)
      
diff --git a/examples/xvslam.py b/examples/xvslam.py
index a3aa590..5a7df33 100755
--- a/examples/xvslam.py
+++ b/examples/xvslam.py
@@ -30,6 +30,8 @@ from xvlidar import XVLidar as Lidar
 
 from pltslamshow import SlamShow
 
+from sys import stdout
+
 if __name__ == '__main__':
 
     # Connect to Lidar unit
@@ -62,8 +64,7 @@ if __name__ == '__main__':
 
         display.setPose(x, y, theta)
 
-        # Exit on ESCape
-        key = display.refresh()
-        if key != None and (key&0x1A):
+        # Exit on window close
+        if not display.refresh():
             exit(0)