Fixed the new code generation for new tasks.