[chore] Fix e2e tests