Note: it turns out we aren't using `mock` in our tests anymore, so I removed this.