Solved fix about get units