From 0b5a63a3d8c6c3aaefc15dcc8c0d8e118d02b94a Mon Sep 17 00:00:00 2001
From: dadav <33197631+dadav@users.noreply.github.com>
Date: Fri, 17 Apr 2020 22:31:30 +0200
Subject: [PATCH] add missing var

---
 pwnagotchi/agent.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/pwnagotchi/agent.py b/pwnagotchi/agent.py
index d24b09e..7fe6a76 100644
--- a/pwnagotchi/agent.py
+++ b/pwnagotchi/agent.py
@@ -307,7 +307,7 @@ class Agent(Client, Automata, AsyncAdvertiser, AsyncTrainer):
 
 
     def start_session_fetcher(self):
-        _thread.start_new_thread(self._fetch_stats, ()) 
+        _thread.start_new_thread(self._fetch_stats, ())
 
 
     def _fetch_stats(self):
@@ -323,7 +323,7 @@ class Agent(Client, Automata, AsyncAdvertiser, AsyncTrainer):
     async def _on_event(self, msg):
         found_handshake = False
         jmsg = json.loads(msg)
-        
+
         if jmsg['tag'] == 'wifi.client.handshake':
             filename = jmsg['data']['file']
             sta_mac = jmsg['data']['station']
@@ -331,6 +331,7 @@ class Agent(Client, Automata, AsyncAdvertiser, AsyncTrainer):
             key = "%s -> %s" % (sta_mac, ap_mac)
             if key not in self._handshakes:
                 self._handshakes[key] = jmsg
+                s = self.session()
                 ap_and_station = self._find_ap_sta_in(sta_mac, ap_mac, s)
                 if ap_and_station is None:
                     logging.warning("!!! captured new handshake: %s !!!", key)