fix(hooks): arguments change error

Fixes #6
garthenweb · Aug 20, 2019 · 0e036bb · 0e036bb
1 parent 5a87c8f
commit 0e036bb
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/lib/hooks.ts b/lib/hooks.ts
@@ -31,12 +31,11 @@ export const useViewportEffect = <T>(
     hasRootProviderAsParent,
   } = useContext(ViewportContext);
 
-  if (!hasRootProviderAsParent) {
-    warnNoContextAvailable('useViewport');
-    return;
-  }
-
   useEffect(() => {
+    if (!hasRootProviderAsParent) {
+      warnNoContextAvailable('useViewport');
+      return;
+    }
     addViewportChangeListener(handleViewportChange, {
       notifyScroll: () => !options.disableScrollUpdates,
       notifyDimensions: () => !options.disableDimensionsUpdates,
@@ -45,7 +44,7 @@ export const useViewportEffect = <T>(
       recalculateLayoutBeforeUpdate: options.recalculateLayoutBeforeUpdate,
     });
     return () => removeViewportChangeListener(handleViewportChange);
-  }, [addViewportChangeListener, removeViewportChangeListener]);
+  }, [addViewportChangeListener || null, removeViewportChangeListener || null]);
 };
 
 export const useViewport = (options: IFullOptions = {}): IViewport => {