[BUGFIX] Fix incompletely refactored update method