Non uniform prior on distance faild with data augmentation

When augmenting data distance is uniformly selected within a range, this does not work when using a different distance prior. needs fix