1 files changed, 3 insertions, 2 deletions
diff --git a/main.py b/main.py
index d13c4ea..fd75b42 100644
--- a/main.py
+++ b/main.py
@@ -9,13 +9,14 @@ import time
 import random
 import gui
 import stats
-from subscriber import DummySubscriber
+from subscriber import EnhancingSubscriber
 from interval_utils import *
 from strategy import *
 
 # global vars
-sub = DummySubscriber()
+sub = EnhancingSubscriber()
 c = client.Client(sub)
+sub.set_client(c)
 stats = stats.Stats()
 
 for i in range(1,10): # 10 connection attempts