Update behat tests to reflect changes to the UI