minor fix in mainUI