Add null check for CaseExpression