Final login bug fix