Fix in test data loading