diff --git a/hatchet/query/engine.py b/hatchet/query/engine.py index 1f0bc7b5..9717e240 100644 --- a/hatchet/query/engine.py +++ b/hatchet/query/engine.py @@ -75,7 +75,7 @@ def _cache_node(self, node, query, dframe): _, filter_func = node_query row = None if isinstance(dframe.index, pd.MultiIndex): - row = pd.concat([dframe.loc[node]], keys=[node], names=["node"]) + row = dframe.xs(node, level="node", drop_level=False) else: row = dframe.loc[node] if filter_func(row):