Fix e2e.
Merge pull request #2802 from brendandburns/fix

Fix e2e.