[fix] Updated e2e tests

@stulees please review. Thanks.