fixing sample tests

- tested to be working on prod