Code review fixes (TUI-54)