Fix up the test to return correct type