Fixing WIFI tests