Minor bug fix in LDA_train