WIP : Fixing path sampler
Fixed to pass tests