Login UI fix