Fixed problem with adding components to units