diff --git a/packages/cms/src/components/Viz/Viz.jsx b/packages/cms/src/components/Viz/Viz.jsx
index 886248d3f..9557c2b6b 100644
--- a/packages/cms/src/components/Viz/Viz.jsx
+++ b/packages/cms/src/components/Viz/Viz.jsx
@@ -186,7 +186,8 @@ Viz.childContextTypes = {
   print: PropTypes.bool,
   updateSource: PropTypes.func,
   variables: PropTypes.object,
-  compVariables: PropTypes.object
+  compVariables: PropTypes.object,
+  comparison: PropTypes.boolean
 };
 
 Viz.contextTypes = {