Small login check fix