Fixes unit test for data directory