Fix to behat tests