Fix wifi issue