From cbafa227b6ec97bd709f41384367dc11a00b1765 Mon Sep 17 00:00:00 2001
From: Pete Shadbolt <hello@peteshadbolt.co.uk>
Date: Sat, 19 Nov 2016 18:52:40 -0800
Subject: [PATCH] Fix up 3D lattice example.

---
 examples/visualization/lattice_3d.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/examples/visualization/lattice_3d.py b/examples/visualization/lattice_3d.py
index a4db00e..45666eb 100644
--- a/examples/visualization/lattice_3d.py
+++ b/examples/visualization/lattice_3d.py
@@ -1,4 +1,4 @@
-from abp.fancy import GraphState
+from abp import GraphState, VizClient
 from abp.util import xyz
 import numpy as np
 import time
@@ -50,5 +50,7 @@ for node in nodes:
 
 for edge in edges:
     psi.act_cz(str(edge[0]), str(edge[1]))
-psi.update()
+
+v = VizClient()
+v.update(psi)