small refactor

Author: Alice Koreman

src/code-view/internal.tsx

@@ -1,6 +1,6 @@
 // Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 // SPDX-License-Identifier: Apache-2.0
-import { Children, createElement, Fragment, isValidElement, ReactElement, useRef } from "react";
+import { Children, createElement, Fragment, ReactElement, useRef } from "react";
 import clsx from "clsx";
 
 import { useCurrentMode } from "@cloudscape-design/component-toolkit/internal";
@@ -48,9 +48,11 @@ export function InternalCodeView({
 
   const regionProps = ariaLabel || ariaLabelledby ? { role: "region" } : {};
 
-  // Create tokenized elements of the content.
+  // Create tokenized React nodes of the content.
   const code = highlight ? highlight(content) : textHighlight(content);
-  const codeElement: ReactElement = isValidElement(code) ? code : createElement(Fragment, null, code);
+  // Create elements from the nodes.
+  const codeElementWrapper: ReactElement = createElement(Fragment, null, code);
+  const codeElement = Children.only(codeElementWrapper.props.children);
 
   return (
     <div