made this extension compatible with behat 3