From d6187b5bfc346fffffe0a92c14e69709a71bad44 Mon Sep 17 00:00:00 2001
From: Kovid Goyal <kovid@kovidgoyal.net>
Date: Mon, 7 Aug 2023 10:25:20 +0530
Subject: [PATCH] ...

---
 kitty_tests/main.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/kitty_tests/main.py b/kitty_tests/main.py
index f012cd63d..d468264ed 100644
--- a/kitty_tests/main.py
+++ b/kitty_tests/main.py
@@ -184,14 +184,13 @@ def run_python_tests(args: Any, go_proc: 'Optional[subprocess.Popen[bytes]]' = N
         sys.stderr.flush()
         print(go_proc.stdout.read().decode('utf-8', 'replace'), end='', flush=True)
         go_proc.stdout.close()
-        go_proc.wait()
+        return go_proc.wait()
 
     if args.module:
         tests = filter_tests_by_module(tests, args.module)
         if not tests._tests:
             if go_proc:
-                print_go()
-                raise SystemExit(go_proc.returncode)
+                raise SystemExit(print_go())
             raise SystemExit('No test module named %s found' % args.module)
 
     if args.name: