Allow state in root modules

spiraldb · Oct 13, 2023 · 4fc9eda · 4fc9eda
1 parent 7abc775
commit 4fc9eda
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/pydust/src/builtins.zig b/pydust/src/builtins.zig
@@ -299,12 +299,14 @@ fn lift(comptime PydustStruct: type) !py.PyObject {
     // Recursively resolve submodules / nested classes
     if (comptime qualName.len > 1) {
         inline for (qualName[1 .. qualName.len - 1]) |part| {
-            defer mod.decref();
+            const prev_mod = mod;
             mod = try mod.get(part);
+            prev_mod.decref();
         }
 
-        defer mod.decref();
+        const prev_mod = mod;
         mod = try mod.get(qualName[qualName.len - 1]);
+        prev_mod.decref();
     }
 
     // Grab the attribute using the final part of the qualified name.