UI: fix a few tests broken by recent commits