diff --git a/mitmaddon/bigsnitch.py b/mitmaddon/bigsnitch.py
index b9184d7..70f0d84 100644
--- a/mitmaddon/bigsnitch.py
+++ b/mitmaddon/bigsnitch.py
@@ -3,7 +3,8 @@
 
 import pdb
 
-from mitmproxy import ctx, Flow
+from mitmproxy import ctx
+from mitmproxy.flow import Flow
 import threading
 from queue import Queue, Empty
 import time
@@ -143,7 +144,6 @@ The network thread communicates with the bigsnitch plugin using zeromq.
 
 """
 
-@dataclass
 class NetworkThread(threading.Thread):
         def __init__(self, name, queue):
                 threading.Thread.__init__(self)