Fixing wifi tests