fixed adding of treatment