PIM-3221: Improve behat