Use the PyModule_*() API instead of manipulating the module dictionary

directly.

1 files changed, 6 insertions, 4 deletions

diff --git a/Modules/arraymodule.c b/Modules/arraymodule.c
index 7372a48834..46de178f68 100644
--- a/Modules/arraymodule.c
+++ b/Modules/arraymodule.c
@@ -1731,12 +1731,14 @@ static PyMethodDef a_methods[] = {
 DL_EXPORT(void)
 initarray(void)
 {
-	PyObject *m, *d;
+	PyObject *m;
 
 	Arraytype.ob_type = &PyType_Type;
 	m = Py_InitModule3("array", a_methods, module_doc);
-	d = PyModule_GetDict(m);
-	PyDict_SetItemString(d, "ArrayType", (PyObject *)&Arraytype);
-	PyDict_SetItemString(d, "array", (PyObject *)&Arraytype);
+
+        Py_INCREF((PyObject *)&Arraytype);
+	PyModule_AddObject(m, "ArrayType", (PyObject *)&Arraytype);
+        Py_INCREF((PyObject *)&Arraytype);
+	PyModule_AddObject(m, "array", (PyObject *)&Arraytype);
 	/* No need to check the error here, the caller will do that */
 }