unit bug fixed