Fixes the UI smoke tests