Thanks Quentin,
Changing the include to uapi/linux/sched/types.h solved the issue.
But just so I'm clear on this - Can you explain when I should use uapi headers vs when to use the regular linux/xxx headers?