UI fix every class