Fix e2e tests

Ah, cool!