Fixed problem with WiFi setup