Bug fix in setting preditctionTests with tree maps