Add missing None check