Fix behat tests