Fix e2e tests (resolve #3)