Fixed update_corpus bug