diff --git a/python/00_simulate_lm_data.py b/python/00_simulate_lm_data.py
index 89d1891..b49247b 100644
--- a/python/00_simulate_lm_data.py
+++ b/python/00_simulate_lm_data.py
@@ -200,12 +200,12 @@
 )
 for i in range(img_shape[2]):
     ax[0, i].imshow(
-        xp.asarray(to_device(img_f1[:, :, i], "cpu")), vmin=0, vmax=vmax, cmap="Greys"
+        np.asarray(to_device(img_f1[:, :, i], "cpu")), vmin=0, vmax=vmax, cmap="Greys"
     )
     ax[0, i].set_title(f"sl {i+1}", fontsize="small")
 
     ax[1, i].imshow(
-        xp.asarray(to_device(img_f2[:, :, i], "cpu")), vmin=0, vmax=vmax, cmap="Greys"
+        np.asarray(to_device(img_f2[:, :, i], "cpu")), vmin=0, vmax=vmax, cmap="Greys"
     )
 
 ax[0, 0].set_ylabel("ground truth frame 1", fontsize="small")
diff --git a/python/01_reconstruct_lm_data.py b/python/01_reconstruct_lm_data.py
index a3fc9f6..9d6317d 100644
--- a/python/01_reconstruct_lm_data.py
+++ b/python/01_reconstruct_lm_data.py
@@ -144,7 +144,7 @@
 fig, ax = plt.subplots(1, recon.shape[2], figsize=(recon.shape[2] * 2, 2))
 for i in range(recon.shape[2]):
     ax[i].imshow(
-        xp.asarray(to_device(recon[:, :, i], "cpu")), vmin=0, vmax=vmax, cmap="Greys"
+        np.asarray(to_device(recon[:, :, i], "cpu")), vmin=0, vmax=vmax, cmap="Greys"
     )
     ax[i].set_title(f"LM recon sl {i+1}", fontsize="small")