[API] Fix update method