Fix for wifi

Wifi fix