Add fix in update service method