Fix some mock return values

This really doesn't affect the tests, but...