py_net: make use of pytalloc_GenericObject_steal()

BUG: https://bugzilla.samba.org/show_bug.cgi?id=12601

Signed-off-by: Stefan Metzmacher <metze@samba.org>
Reviewed-by: Andrew Bartlett <abartlet@samba.org>

1 files changed, 2 insertions, 3 deletions

diff --git a/source4/libnet/py_net.c b/source4/libnet/py_net.c
index 3e70c79ea74..5180c247a71 100644
--- a/source4/libnet/py_net.c
+++ b/source4/libnet/py_net.c
@@ -457,7 +457,7 @@ static PyObject *py_net_replicate_init(py_net_Object *self, PyObject *args, PyOb
 	s->chunk.forest = &s->forest;
 	s->chunk.dest_dsa = &s->dest_dsa;
 
-	return pytalloc_CObject_FromTallocPtr(s);
+	return pytalloc_GenericObject_steal(s);
 }
 
 
@@ -485,9 +485,8 @@ static PyObject *py_net_replicate_chunk(py_net_Object *self, PyObject *args, PyO
 		return NULL;
 	}
 
-	s = talloc_get_type(PyCObject_AsVoidPtr(py_state), struct replicate_state);
+	s = pytalloc_get_type(py_state, struct replicate_state);
 	if (!s) {
-		PyErr_SetString(PyExc_TypeError, "Expected replication_state");
 		return NULL;
 	}