Login controller fix