Bugfix

Small bug fix on `.update()` method