correction on unit entity behaviors