Update our tests to the new ref return syntax