private Reference<LivingEntity> target_ref;

Then when setting reference use the regular living entity in method still.

this.target_ref = new SoftReference(target);