small UI correction