summaryrefslogtreecommitdiff
path: root/src/runtime/python/pypgf.c
diff options
context:
space:
mode:
Diffstat (limited to 'src/runtime/python/pypgf.c')
-rw-r--r--src/runtime/python/pypgf.c7
1 files changed, 4 insertions, 3 deletions
diff --git a/src/runtime/python/pypgf.c b/src/runtime/python/pypgf.c
index eb6725050..5adf91c10 100644
--- a/src/runtime/python/pypgf.c
+++ b/src/runtime/python/pypgf.c
@@ -920,9 +920,10 @@ Iter_fetch_token(IterObject* self)
if (tp == NULL)
return NULL;
- PyObject* ty_tok = gu2py_string(tp->tok);
- PyObject* res = Py_BuildValue("(f,O)", tp->prob, ty_tok);
- Py_DECREF(ty_tok);
+ PyObject* py_tok = gu2py_string(tp->tok);
+ PyObject* py_cat = gu2py_string(tp->cat);
+ PyObject* res = Py_BuildValue("(f,O,O)", tp->prob, py_tok, py_cat);
+ Py_DECREF(py_tok);
return res;
}