Fixes to e2e script