Fixing missed monkey patch