diff --git a/src/hooks/useViewport.ts b/src/hooks/useViewport.ts
index 3575b8bf..f4e61587 100644
--- a/src/hooks/useViewport.ts
+++ b/src/hooks/useViewport.ts
@@ -1,3 +1,5 @@
+import shallow from 'zustand/shallow';
+
 import { useStore } from '../store';
 import { Viewport, ReactFlowState } from '../types';
 
@@ -8,7 +10,7 @@ const viewportSelector = (state: ReactFlowState) => ({
 });
 
 function useViewport(): Viewport {
-  const viewport = useStore(viewportSelector);
+  const viewport = useStore(viewportSelector, shallow);
 
   return viewport;
 }