Fix the generation of discourse trees

Signed-off-by: Marius L. Jøhndal <mariuslj@ifi.uio.no>