Quick UI fix