Unit add bug fixed