Improve protractor integration