Add check for null value