Fix tests for the main update service.