Fix e2e test cases