Updating tests to reflect the location of the method