Fixes issue with task reassign flow