further changes to work with protractor