Fix directory problem