Monkey patch correct class