Fix Behat tests