From 7d6faba644151c33128691bdb96131ce2287c2dd Mon Sep 17 00:00:00 2001
From: Noe Brucy <noe.brucy@cea.fr>
Date: Wed, 28 Apr 2021 19:21:09 +0200
Subject: [PATCH] [pspec] correct keys

---
 pspec_new.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pspec_new.py b/pspec_new.py
index 0d3f773..6299922 100644
--- a/pspec_new.py
+++ b/pspec_new.py
@@ -838,7 +838,7 @@ def main(arg):
         # 2D power spectra -------------------------------------------------------------
         ns = 2 ** clvl
         f = "_%%(i)0%dd" % (np.floor(np.log10(ns)) + 1)
-        for v in var_names:
+        for v in list(pcubes2.keys()):
             for i in range(ns):
                 pspec, kbins, pspec2, fbins = pspectrum(
                     pcubes2[v][:, :, i], cubes_k["k"][:, :, i], kbins, knorm, arg.fbins