From af9a3d933b8aa03726d22b6ae205b6d4838c3147 Mon Sep 17 00:00:00 2001
From: Axel Kohlmeyer <akohlmey@gmail.com>
Date: Tue, 24 Jun 2025 22:15:04 -0400
Subject: [PATCH] make consistent for both branches

---
 python/lammps/pylammps.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/python/lammps/pylammps.py b/python/lammps/pylammps.py
index f19b2ce78c..56a82b7433 100644
--- a/python/lammps/pylammps.py
+++ b/python/lammps/pylammps.py
@@ -132,6 +132,7 @@ class AtomList(object):
     if index not in self._loaded:
       if self.dimensions == 2:
         atom = Atom2D(self._pylmp, index)
+        self._loaded[index] = atom
       else:
         atom = Atom(self._pylmp, index)
         self._loaded[index] = atom