Updated tests to work with proractor