From 637b31cf446f35fd958d07a69b053203c3d24cf7 Mon Sep 17 00:00:00 2001
From: rstiskalek <richard.stiskalek@protonmail.com>
Date: Thu, 19 Oct 2023 13:57:15 +0100
Subject: [PATCH] Fix key bug

---
 csiborgtools/read/readsim.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/csiborgtools/read/readsim.py b/csiborgtools/read/readsim.py
index 88cee8e..3bf36b0 100644
--- a/csiborgtools/read/readsim.py
+++ b/csiborgtools/read/readsim.py
@@ -522,9 +522,9 @@ class QuijoteReader(BaseReader):
         vel = fof.GroupVel * (1 + self.read_info(nsnap, nsim)["redshift"])
         for i, p in enumerate(["x", "y", "z"]):
             data[p] = pos[:, i]
-            data["fof_v" + p] = vel[:, i]
+            data[f"v{p}"] = vel[:, i]
         data["group_mass"] = fof.GroupMass * 1e10
-        data["fof_npart"] = fof.GroupLen
+        data["npart"] = fof.GroupLen
         # We want to start indexing from 1. Index 0 is reserved for
         # particles unassigned to any FoF group.
         data["index"] = 1 + numpy.arange(data.size, dtype=numpy.int32)