Make small UI fix