Fix login UI