Only load integration_test dataset in testing