Allow use of map in with-stub to specify return values