fix remaining e2e tests