Updated the code for latest changes done in Behat