Fixes broken e2e test