Fix WiFi test (Xia Wang)

Change-Id: I50fc7ea75b4e55325a02341b2c64db430d709e98