Fix the test data source