Fix data directory issue