adapt extension to behat 3.0