Fixes the UI smoke tests

:+1: