Add missing nil check