Add missing 'LoginFunction' handle logic