Fix data provider so tests are running again