COMP: Fixed code to match with changed in vtkMRMLSelectionNode API.