Simple UI fix.