Fixed the e2e test