Fix to monkey patch test