Update to make generated files compile