diff --git a/mlir/lib/Bindings/Python/IRModules.h b/mlir/lib/Bindings/Python/IRModules.h
--- a/mlir/lib/Bindings/Python/IRModules.h
+++ b/mlir/lib/Bindings/Python/IRModules.h
@@ -317,12 +317,12 @@
   void contextExit(pybind11::object excType, pybind11::object excVal,
                    pybind11::object excTb);
 
-  /// Gets a capsule wrapping the void* within the MlirContext.
+  /// Gets a capsule wrapping the void* within the MlirLocation.
   pybind11::object getCapsule();
 
-  /// Creates a PyMlirContext from the MlirContext wrapped by a capsule.
-  /// Note that PyMlirContext instances are uniqued, so the returned object
-  /// may be a pre-existing object. Ownership of the underlying MlirContext
+  /// Creates a PyLocation from the MlirLocation wrapped by a capsule.
+  /// Note that PyLocation instances are uniqued, so the returned object
+  /// may be a pre-existing object. Ownership of the underlying MlirLocation
   /// is taken by calling this function.
   static PyLocation createFromCapsule(pybind11::object capsule);
 
@@ -597,12 +597,12 @@
   operator MlirAttribute() const { return attr; }
   MlirAttribute get() const { return attr; }
 
-  /// Gets a capsule wrapping the void* within the MlirContext.
+  /// Gets a capsule wrapping the void* within the MlirAttribute.
   pybind11::object getCapsule();
 
-  /// Creates a PyMlirContext from the MlirContext wrapped by a capsule.
-  /// Note that PyMlirContext instances are uniqued, so the returned object
-  /// may be a pre-existing object. Ownership of the underlying MlirContext
+  /// Creates a PyAttribute from the MlirAttribute wrapped by a capsule.
+  /// Note that PyAttribute instances are uniqued, so the returned object
+  /// may be a pre-existing object. Ownership of the underlying MlirAttribute
   /// is taken by calling this function.
   static PyAttribute createFromCapsule(pybind11::object capsule);
 
@@ -643,12 +643,12 @@
   operator MlirType() const { return type; }
   MlirType get() const { return type; }
 
-  /// Gets a capsule wrapping the void* within the MlirContext.
+  /// Gets a capsule wrapping the void* within the MlirType.
   pybind11::object getCapsule();
 
-  /// Creates a PyMlirContext from the MlirContext wrapped by a capsule.
-  /// Note that PyMlirContext instances are uniqued, so the returned object
-  /// may be a pre-existing object. Ownership of the underlying MlirContext
+  /// Creates a PyType from the MlirType wrapped by a capsule.
+  /// Note that PyType instances are uniqued, so the returned object
+  /// may be a pre-existing object. Ownership of the underlying MlirType
   /// is taken by calling this function.
   static PyType createFromCapsule(pybind11::object capsule);