Patch correct post generate classes