diff --git a/system/ui/text.py b/system/ui/text.py
index de25688c6d..23d2184459 100755
--- a/system/ui/text.py
+++ b/system/ui/text.py
@@ -84,6 +84,10 @@ class TextWindow:
     while self._renderer is None and self._thread.is_alive():
       time.sleep(0.01)
 
+  def wait_for_exit(self):
+    while self._thread.is_alive():
+      time.sleep(0.01)
+
   def _run(self):
     if os.getenv("CI") is not None:
       return