diff --git a/tools/lib/uflow.py b/tools/lib/uflow.py
index 02cad55..63fab87 100755
--- a/tools/lib/uflow.py
+++ b/tools/lib/uflow.py
@@ -206,4 +206,7 @@
 
 bpf["calls"].open_perf_buffer(print_event)
 while 1:
-    bpf.perf_buffer_poll()
+    try:
+        bpf.perf_buffer_poll()
+    except KeyboardInterrupt:
+        exit()