diff --git a/pymc3/tests/test_data_container.py b/pymc3/tests/test_data_container.py
index 9db76754b44..487dccf5ec8 100644
--- a/pymc3/tests/test_data_container.py
+++ b/pymc3/tests/test_data_container.py
@@ -101,5 +101,5 @@ def test_model_to_graphviz_for_model_with_data_container(self):
             pm.sample(1000, init=None, tune=1000, chains=1)
 
         g = pm.model_to_graphviz(model)
-        text = 'x [label="x ~ Deterministic" shape=box style=filled]'
+        text = 'x [label="x ~ Data" shape=box style=filled]'
         assert text in g.source