Update to fix issue generating code