Fix for wifi test
Merge pull request #798 from AndreiH/fix_wifi_test

Fix for wifi test