Fixes the UI smoke tests

I will check if, we have a more cleaner method to shorten it up. But for now this is what I could think of.