Minor test tweak to align return types.