Fix to test data directory