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)