Fix for wifi test