Set task stage upon creation/editing