Fix EndToEnd test
Fix EndToEnd test