test: Fix tests not passing.

We changed the interface to be more generic, by taking a website as an
argument, but the test was not updated.