TransformerAgents - error running example notebook

When running the provided example notebook Google Colab

Got the following error when running a more complex query“In which city the wife of Barack Obama was born?”)

ValueError: expected sequence of length 31385 at dim 1 (got 31386)
The above exception was the direct cause of the following exception: