Fixed failing e2e test