diff --git a/src/main/scala/com/github/shadowsocks/acl/CustomRulesFragment.scala b/src/main/scala/com/github/shadowsocks/acl/CustomRulesFragment.scala index 01cf8588..76aa3907 100644 --- a/src/main/scala/com/github/shadowsocks/acl/CustomRulesFragment.scala +++ b/src/main/scala/com/github/shadowsocks/acl/CustomRulesFragment.scala @@ -148,7 +148,13 @@ class CustomRulesFragment extends ToolbarFragment with Toolbar.OnMenuItemClickLi def remove(i: Int) { val j = i - acl.subnets.size - if (j < 0) acl.subnets.remove(i) else acl.proxyHostnames.remove(j) + if (j < 0) { + undoManager.remove((i, acl.subnets(i))) + acl.subnets.remove(i) + } else { + undoManager.remove((j, acl.proxyHostnames(j))) + acl.proxyHostnames.remove(j) + } notifyItemRemoved(i) apply() }