Tried to fix update method