Update functionality fix