Fixing e2e unit tests