diff --git a/examples/run_wela.py b/examples/run_wela.py
index f8b36563bf70cefedaf3d914c627dc56b86c6b9b..91cd0f374bd4bbd5f87183caa42ad1dc2caec1d3 100644
--- a/examples/run_wela.py
+++ b/examples/run_wela.py
@@ -19,7 +19,7 @@ server_info = {
     "username": "upload",
     "password": "gothamc1ty",
 }
-view = True
+view = False
 # pick the experiment to analyse
 omid = 469  # expt_id e.g., 469
 
@@ -44,8 +44,9 @@ if view:
     sys.exit(0)
 else:
     # run dataloader
-    key_index = "median_GFP"
+    key_index = "median_GFP_Z"
     dl = dataloader(h5dir, ".")
+    dl.define_g2a_dict(["GFP_Z"])
     expt = [omid_full for omid_full in omids if str(omid) in omid_full][0]
     dl.load(expt, key_index=key_index, cutoff=0.8)
     dl.save()