Add e2e test on main
E2e test fix