Adjust test to work in Phantom.js