Wifi fix

Change-Id: I516ccb8d91c40f7a666e87e5872803c1b14344aa