can we also add an `assert_equals` on the return value?