Just added a couple lines to the UI Test for the new buttons.