Updated End to End test with the new API