Fixes the UI smoke tests

Yeah..this needs to be updated..I'll see if I can come out with a clean way..