Monkey patch before migration