Fix plotting with row number as index