Improve behat context initialization