diff --git a/fdisk.c b/fdisk.c
index 89e4cff..c44b2f1 100644
--- a/fdisk.c
+++ b/fdisk.c
@@ -65,6 +65,7 @@ PyInit_fdisk(void)
 
 	Context_AddModuleObject(m);
 	Label_AddModuleObject(m);
+	Partition_AddModuleObject(m);
 
 
 	return m;
diff --git a/fdisk.h b/fdisk.h
index 9e01baf..496b6f7 100644
--- a/fdisk.h
+++ b/fdisk.h
@@ -44,6 +44,7 @@ extern PyTypeObject PartitionType;
 
 extern void Context_AddModuleObject(PyObject *mod);
 extern void Label_AddModuleObject(PyObject *mod);
+extern void Partition_AddModuleObject(PyObject *mod);
 
 extern PyObject *PyObjectResultStr(const char *s);
 extern PyObject *PyObjectResultLabel(struct fdisk_label *lb);