Add a workaround for word_id