UI - fix

Signed-off-by: Thomas Yuill <thomas.yuill@gmail.com>