2 files changed, 16 insertions, 11 deletions
diff --git a/test/scripts/all.py b/test/scripts/all.py
index e6da7e51..7d30d757 100644
--- a/test/scripts/all.py
+++ b/test/scripts/all.py
@@ -3,6 +3,10 @@ def clientconnect(ctx, cc):
     ctx.log("XCLIENTCONNECT")
     log.append("clientconnect")
 
+def serverconnect(ctx, cc):
+    ctx.log("XSERVERCONNECT")
+    log.append("serverconnect")
+
 def request(ctx, r):
     ctx.log("XREQUEST")
     log.append("request")
diff --git a/test/scripts/concurrent_decorator.py b/test/scripts/concurrent_decorator.py
index c1c2651e..8e132006 100644
--- a/test/scripts/concurrent_decorator.py
+++ b/test/scripts/concurrent_decorator.py
@@ -1,6 +1,17 @@
 import time
 from libmproxy.script import concurrent
 
+
+@concurrent
+def clientconnect(context, cc):
+    context.log("clientconnect")
+
+
+@concurrent
+def serverconnect(context, sc):
+    context.log("serverconnect")
+
+
 @concurrent
 def request(context, flow):
     time.sleep(0.1)
@@ -17,15 +28,5 @@ def error(context, err):
 
 
 @concurrent
-def clientconnect(context, cc):
-    context.log("clientconnect")
-
-
-@concurrent
 def clientdisconnect(context, dc):
-    context.log("clientdisconnect")
-
-
-@concurrent
-def serverconnect(context, sc):
-    context.log("serverconnect")
-\ No newline at end of file
+    context.log("clientdisconnect")
+\ No newline at end of file