Oops, forgot to change the function return values in the test