Fix up for ProBAM updates