Fixed broken end to end tests