chore: refactor

cognitedata · Aug 1, 2023 · 9ec8db5 · 9ec8db5
1 parent c025a1e
commit 9ec8db5
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/react-components/stories/HighlightNode.stories.tsx b/react-components/stories/HighlightNode.stories.tsx
@@ -78,16 +78,12 @@ const StoryContent = ({
   resources: AddResourceOptions[];
   fdmAssetMappingConfig: FdmAssetMappingsConfig;
 }) => {
-  const viewer = useReveal();
-
   const [nodeData, setNodeData] = useState<any>();
 
   const callback = (nodeData: NodeDataResult<any>) => {
     setNodeData(nodeData.data);
 
-    if (!(viewer.models[0] instanceof CogniteCadModel)) return;
-
-    viewer.models[0].assignStyledNodeCollection(
+    nodeData.model.assignStyledNodeCollection(
       new TreeIndexNodeCollection([nodeData.cadNode.treeIndex]),
       DefaultNodeAppearance.Highlighted
     );