PIM-3377: Improve behat