Fix generative story bug in DPMM