Fixed chat e2e test suite

Ignore this change for now. I'll fix it in a follow-up check-in.