diff --git a/enrich.py b/enrich.py index da0c0a6..c056516 100755 --- a/enrich.py +++ b/enrich.py @@ -155,11 +155,13 @@ def get_aps_on_channel(capture, ap_channel): print(f"[!] Could not parse channel number: {ap_channel}") return 0 - from_channel_freq = get_channel_from_freq(ap_channel) - if not from_channel_freq: + channel_info = get_freq_details(ap_channel) + if not channel_info: print(f"[!] Invalid channel: {ap_channel}") return 0 + from_channel_freq = channel_info["freq"] + aps = set() for packet in capture: