Fix broken test (TUI-129)