tiny little UI fix