Updated e2e to reflect changes in UI