Fix to behat context