Attempt to fix probability mapping with probtrackx