Update e2e_test.go