fixing sample tests

- tested to be working on prod
Doesn't look like we're using this.