@@ -647,7 +647,10 @@ impl PyObject {
     }
 
     pub fn set_class(&self, typ: PyTypeRef, vm: &VirtualMachine) {
-        self.0.typ.swap_to_frame(typ, vm);
+        let old = unsafe { self.0.typ.swap(typ) };
+        if let Some(frame) = vm.current_frame() {
+            frame.temperary_refs.lock().push(old.into());
+        }
     }
 
     #[inline(always)]
 
@@ -250,18 +250,15 @@ impl<T: PyObjectPayload> Deref for PyAtomicRef<T> {
 }
 
 impl<T: PyObjectPayload> PyAtomicRef<T> {
-    unsafe fn swap(&self, pyref: PyRef<T>) -> PyRef<T> {
+    /// # Safety
+    /// The caller is responsible to keep the returned PyRef alive
+    /// until no more reference can be used via PyAtomicRef::deref()
+    #[must_use]
+    pub unsafe fn swap(&self, pyref: PyRef<T>) -> PyRef<T> {
         let py = PyRef::leak(pyref);
         let old = self.0.swap(NonNull::from(py));
         PyRef::from_raw(old.as_ptr())
     }
-
-    pub fn swap_to_frame(&self, pyref: PyRef<T>, vm: &VirtualMachine) {
-        let old = unsafe { self.swap(pyref) };
-        if let Some(frame) = vm.current_frame() {
-            frame.temperary_refs.lock().push(old.into());
-        }
-    }
 }
 
 pub trait AsObject
Original file line number	Diff line number	Diff line change
`@@ -647,7 +647,10 @@ impl PyObject {`
`647`	`647`	`}`
`648`	`648`
`649`	`649`	`pub fn set_class(&self, typ: PyTypeRef, vm: &VirtualMachine) {`
`650`		`- self.0.typ.swap_to_frame(typ, vm);`
	`650`	`+ let old = unsafe { self.0.typ.swap(typ) };`
	`651`	`+ if let Some(frame) = vm.current_frame() {`
	`652`	`+ frame.temperary_refs.lock().push(old.into());`
	`653`	`+ }`
`651`	`654`	`}`
`652`	`655`
`653`	`656`	`#[inline(always)]`