Fix directory setup