Updating tests to explicitly rely on the mock.