bug fixed in test_toy_corpus