diff --git a/property.go b/property.go index f1c80b9..25afced 100644 --- a/property.go +++ b/property.go @@ -167,7 +167,7 @@ func (s *Selection) RemoveClass(class ...string) *Selection { } else { classes, attr := getClassesAndAttr(n, true) for _, rcl := range rclasses { - classes = strings.Replace(classes, " "+rcl+" ", " ", -1) + classes = strings.ReplaceAll(classes, " "+rcl+" ", " ") } setClasses(n, attr, classes) @@ -191,8 +191,9 @@ func (s *Selection) ToggleClass(class ...string) *Selection { for _, n := range s.Nodes { classes, attr := getClassesAndAttr(n, true) for _, tcl := range tcls { - if strings.Contains(classes, " "+tcl+" ") { - classes = strings.Replace(classes, " "+tcl+" ", " ", -1) + spaceAroundTcl := " " + tcl + " " + if strings.Contains(classes, spaceAroundTcl) { + classes = strings.ReplaceAll(classes, spaceAroundTcl, " ") } else { classes += tcl + " " }