Fix doctest failure with dataframe display