PR Fixup: Add UI Test