Fixed chat e2e test suite

Please ignore for now. I will fix this in a follow-up check-in.