feat(API): Add safety to unit modification