Fix the test to match expected returned value