Fix e2e tests
Merge pull request #2017 from jbeda/fix-e2e

Fix e2e tests