Fix wifi bug