X-Git-Url: https://git.distorted.org.uk/~mdw/catacomb-python/blobdiff_plain/d050e0fa0f1cad4b77fc11240990068d4a13da49..f52568b48a26bce7f62c3c03402e9f705f16332b:/buffer.c

diff --git a/buffer.c b/buffer.c
index b15a245..ac3e56b 100644
--- a/buffer.c
+++ b/buffer.c
@@ -446,8 +446,7 @@ static PyObject *wbmeth_putecpt(PyObject *me, PyObject *arg)
 {
   ec pt = EC_INIT;
   if (!PyArg_ParseTuple(arg, "O&:putecpt", convecpt, &pt)) return (0);
-  if (EC_ATINF(&pt)) ensure(me, 2);
-  else ensure(me, 4 + mp_octets(pt.x) + mp_octets(pt.y));
+  ensure(me, EC_ATINF(&pt) ? 2 : 6 + mp_octets(pt.x) + mp_octets(pt.y));
   buf_putec(BUF_B(me), &pt); assert(BOK(BUF_B(me)));
   EC_DESTROY(&pt);
   RETURN_ME;