diff --git a/neotest_python/params_getter.py b/neotest_python/params_getter.py index 8f0ba13..1f21eab 100644 --- a/neotest_python/params_getter.py +++ b/neotest_python/params_getter.py @@ -136,7 +136,7 @@ async def serve_socket(): if python_socket_path.exists(): print(python_socket_path) - return + atexit.register(lambda: _close_socket(python_socket_path)) signal.signal(signal.SIGTERM, lambda x, _: _close_socket(python_socket_path))