Fix WiFi build