diff --git a/fdtd.py b/fdtd.py
index 37f70f9..e718177 100644
--- a/fdtd.py
+++ b/fdtd.py
@@ -228,7 +228,7 @@ def main():
 
     with lzma.open('saved_simulation', 'wb') as f:
         def unvec(f):
-            return meanas.unvec(f, grid.shape)
+            return meanas.fdmath.unvec(f, grid.shape)
         d = {
             'grid': grid,
             'E': unvec(sim.E.get()),
diff --git a/opencl_fdtd/simulation.py b/opencl_fdtd/simulation.py
index 381047f..8023d81 100644
--- a/opencl_fdtd/simulation.py
+++ b/opencl_fdtd/simulation.py
@@ -48,7 +48,7 @@ class Simulation(object):
             event.wait()
 
             with lzma.open('saved_simulation', 'wb') as f:
-                dill.dump(fdfd_tools.unvec(sim.E.get(), grid.shape), f)
+                dill.dump(meanas.fdmath.unvec(sim.E.get(), grid.shape), f)
 
     Code in the form
         event2 = sim.update_H([event0, event1])