diff --git a/examples/wifi/iperf/main/cmd_wifi.c b/examples/wifi/iperf/main/cmd_wifi.c
index a6be001c9b..d59b7c878a 100644
--- a/examples/wifi/iperf/main/cmd_wifi.c
+++ b/examples/wifi/iperf/main/cmd_wifi.c
@@ -61,6 +61,11 @@ static void scan_done_handler(void* arg, esp_event_base_t event_base,
     wifi_ap_record_t *ap_list_buffer;
 
     esp_wifi_scan_get_ap_num(&sta_number);
+    if (!sta_number) {
+        ESP_LOGE(TAG, "No AP found");
+        return;
+    }
+
     ap_list_buffer = malloc(sta_number * sizeof(wifi_ap_record_t));
     if (ap_list_buffer == NULL) {
         ESP_LOGE(TAG, "Failed to malloc buffer to print scan results");