Missing check if it has task_generator
Merge pull request #4478 from juztas/missing_check

Missing check if it has task_generator