Fix update error in Work controller