Fixed AddUnit bug