diff --git a/packages/core/src/container/ReactFlow/Wrapper.tsx b/packages/core/src/container/ReactFlow/Wrapper.tsx
index b6069948..f0f56d24 100644
--- a/packages/core/src/container/ReactFlow/Wrapper.tsx
+++ b/packages/core/src/container/ReactFlow/Wrapper.tsx
@@ -1,10 +1,13 @@
-import { FC, PropsWithChildren, useContext } from 'react';
+import { useContext } from 'react';
+import type { FC, PropsWithChildren } from 'react';
 
 import StoreContext from '../../contexts/RFStoreContext';
 import ReactFlowProvider from '../../components/ReactFlowProvider';
 
 const Wrapper: FC<PropsWithChildren> = ({ children }) => {
-  if (useContext(StoreContext)) {
+  const isWrapped = useContext(StoreContext);
+
+  if (isWrapped) {
     // we need to wrap it with a fragment because it's not allowed for children to be a ReactNode
     // https://github.com/DefinitelyTyped/DefinitelyTyped/issues/18051
     return <>{children}</>;