Fix python path
Checkpoint