Fix dir struct
checkpoint