diff --git a/public/app/features/alerting/unified/components/rule-editor/RuleEditorSection.tsx b/public/app/features/alerting/unified/components/rule-editor/RuleEditorSection.tsx
index 91ccead570e..8c872e2b4f7 100644
--- a/public/app/features/alerting/unified/components/rule-editor/RuleEditorSection.tsx
+++ b/public/app/features/alerting/unified/components/rule-editor/RuleEditorSection.tsx
@@ -39,9 +39,7 @@ export const RuleEditorSection = ({
                 <InlineSwitch
                   id={`advanced-switch-${stepNo}`}
                   data-testid={
-                    switchMode.isAdvancedMode
-                      ? 'query-and-expressions-advanced-options'
-                      : 'query-and-expressions-simple-options'
+                    switchMode.isAdvancedMode ? `advanced-switch-${stepNo}-advanced` : `advanced-switch-${stepNo}-basic`
                   }
                   value={switchMode.isAdvancedMode}
                   onChange={(event) => {