diff --git a/platform/ui/src/contextProviders/ViewportGridProvider.jsx b/platform/ui/src/contextProviders/ViewportGridProvider.jsx
index b718d5145..480114d61 100644
--- a/platform/ui/src/contextProviders/ViewportGridProvider.jsx
+++ b/platform/ui/src/contextProviders/ViewportGridProvider.jsx
@@ -139,3 +139,8 @@ ViewportGridProvider.propTypes = {
 };
 
 export const useViewportGrid = () => useContext(ViewportGridContext);
+
+// TODO:
+// - Update dependents of `useViewportGrid` to use new API instead of generic `dispatch`
+// - Update `@ohif/core` interface `setServiceImplementation` to accomodate defined API methods
+// - Update `@ohif/ext-cornerstone` commands to use new service to get the activeViewportIndex and restore command functionality