fix(todo): Fixing issue with todos not handling generator functions