Please do add a unit test to assert that the return value is correct too.