Fix status code issue for e2e tests