diff --git a/vm/src/builtins/function.rs b/vm/src/builtins/function.rs
index 44dadf340..bf1451c92 100644
--- a/vm/src/builtins/function.rs
+++ b/vm/src/builtins/function.rs
@@ -2,7 +2,7 @@
 mod jitfunc;
 
 use super::{
-    tuple::PyTupleTyped, PyAsyncGen, PyCode, PyCoroutine, PyDictRef, PyGenerator, PyStrRef,
+    tuple::PyTupleTyped, PyAsyncGen, PyCode, PyCoroutine, PyDictRef, PyGenerator, PyStr, PyStrRef,
     PyTupleRef, PyTypeRef,
 };
 use crate::common::lock::PyMutex;
@@ -21,9 +21,6 @@ use crate::{common::lock::OnceCell, convert::ToPyObject};
 use itertools::Itertools;
 #[cfg(feature = "jit")]
 use rustpython_jit::CompiledCode;
-use rustpython_vm::builtins::PyStr;
-
-pub type PyFunctionRef = PyRef<PyFunction>;
 
 #[pyclass(module = false, name = "function")]
 #[derive(Debug)]
diff --git a/vm/src/stdlib/builtins.rs b/vm/src/stdlib/builtins.rs
index 33513b9ee..e1e785c73 100644
--- a/vm/src/stdlib/builtins.rs
+++ b/vm/src/stdlib/builtins.rs
@@ -14,7 +14,7 @@ mod builtins {
         builtins::{
             asyncgenerator::PyAsyncGen,
             enumerate::PyReverseSequenceIterator,
-            function::{PyCellRef, PyFunctionRef},
+            function::{PyCellRef, PyFunction},
             int::PyIntRef,
             iter::PyCallableIterator,
             list::{PyList, SortOptions},
@@ -802,7 +802,7 @@ mod builtins {
 
     #[pyfunction]
     pub fn __build_class__(
-        function: PyFunctionRef,
+        function: PyRef<PyFunction>,
         qualified_name: PyStrRef,
         bases: PosArgs,
         mut kwargs: KwArgs,