Assert the return value in the test