Updated to use `stub` helper.