Update the Test to use the service