Able to reproduce error on smaller dataset