Minor UI correction