Fixed issue with edit task