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