UI: fixing tests for #133