Fix protractor tests