diff --git a/src/viewer/rules/Dialogs/FlagsDialog.tsx b/src/viewer/rules/Dialogs/FlagsDialog.tsx
index d6b3eec..0a5a50a 100644
--- a/src/viewer/rules/Dialogs/FlagsDialog.tsx
+++ b/src/viewer/rules/Dialogs/FlagsDialog.tsx
@@ -101,7 +101,7 @@ export default class FlagsDialog extends React.Component<Props, State> {
 			const rule = this.state.rules[ruleIndex].reset();
 			this.updateRule(rule, ruleIndex);
 		}
-		this.setState({ showEdit: false }, () => {
+		this.setState({ selectedRule: -1, showEdit: false }, () => {
 			if (isClose === true) this.props.onClose(this.state.rules);
 			else this.props.onReturn(this.state.rules);
 		});
diff --git a/src/viewer/rules/Dialogs/StatesDialog.tsx b/src/viewer/rules/Dialogs/StatesDialog.tsx
index e8410d2..b032d94 100644
--- a/src/viewer/rules/Dialogs/StatesDialog.tsx
+++ b/src/viewer/rules/Dialogs/StatesDialog.tsx
@@ -88,7 +88,7 @@ export default class StatesDialog extends React.Component<Props, State> {
 			const rule = this.state.rules[ruleIndex].reset();
 			this.updateRule(rule, ruleIndex);
 		}
-		this.setState({ showEdit: false }, () => {
+		this.setState({ selectedRule: -1, showEdit: false }, () => {
 			if (isClose === true) this.props.onClose(this.state.rules);
 			else this.props.onReturn(this.state.rules);
 		});