[Generator] Fix integration problem