diff --git a/scripts/98_profiler.py b/scripts/98_profiler.py
index 91506f39..af7c1d4a 100644
--- a/scripts/98_profiler.py
+++ b/scripts/98_profiler.py
@@ -1,11 +1,11 @@
-from pkpy import *
 import time
 
 def costly_func(n: int):
-    time.sleep(n)
+    time.sleep(n)               # 2s
+    time.sleep(1)               # 1s
 
 x = 1
 y = 2
-costly_func(2)
+costly_func(2)      # 3s
 
-time.sleep(1)
+time.sleep(1)       # 1s