Wifi fix

Change-Id: I06b799b491d17a4f9fed6202e985597de041608f