Fix directory issue