diff --git a/x-pack/plugins/ingest_pipelines/public/application/components/pipeline_processors_editor/components/test_pipeline/documents_dropdown/documents_dropdown.tsx b/x-pack/plugins/ingest_pipelines/public/application/components/pipeline_processors_editor/components/test_pipeline/documents_dropdown/documents_dropdown.tsx
index 9c8d6c8bdb289..19c3c49396c6e 100644
--- a/x-pack/plugins/ingest_pipelines/public/application/components/pipeline_processors_editor/components/test_pipeline/documents_dropdown/documents_dropdown.tsx
+++ b/x-pack/plugins/ingest_pipelines/public/application/components/pipeline_processors_editor/components/test_pipeline/documents_dropdown/documents_dropdown.tsx
@@ -27,7 +27,7 @@ const i18nTexts = {
       defaultMessage: 'Documents:',
     }
   ),
-  addDocumentsButtonLabel: i18n.translate(
+  editDocumentsButtonLabel: i18n.translate(
     'xpack.ingestPipelines.pipelineEditor.testPipeline.documentsDropdown.editDocumentsButtonLabel',
     {
       defaultMessage: 'Edit documents',
@@ -124,7 +124,7 @@ export const DocumentsDropdown: FunctionComponent<Props> = ({
           }}
           data-test-subj="editDocumentsButton"
         >
-          {i18nTexts.addDocumentsButtonLabel}
+          {i18nTexts.editDocumentsButtonLabel}
         </EuiButton>
       </EuiPopoverFooter>
     </EuiPopover>
diff --git a/x-pack/plugins/ingest_pipelines/public/application/components/pipeline_processors_editor/components/test_pipeline/test_pipeline_actions.tsx b/x-pack/plugins/ingest_pipelines/public/application/components/pipeline_processors_editor/components/test_pipeline/test_pipeline_actions.tsx
index 83a9303859d2a..60cfd882e1d82 100644
--- a/x-pack/plugins/ingest_pipelines/public/application/components/pipeline_processors_editor/components/test_pipeline/test_pipeline_actions.tsx
+++ b/x-pack/plugins/ingest_pipelines/public/application/components/pipeline_processors_editor/components/test_pipeline/test_pipeline_actions.tsx
@@ -88,6 +88,7 @@ export const TestPipelineActions: FunctionComponent = () => {
       {openTestPipelineFlyout && (
         <TestPipelineFlyout
           activeTab={activeFlyoutTab}
+          setActiveTab={setActiveFlyoutTab}
           processors={{
             processors: processors.state.processors,
             onFailure: processors.state.onFailure,
diff --git a/x-pack/plugins/ingest_pipelines/public/application/components/pipeline_processors_editor/components/test_pipeline/test_pipeline_flyout.container.tsx b/x-pack/plugins/ingest_pipelines/public/application/components/pipeline_processors_editor/components/test_pipeline/test_pipeline_flyout.container.tsx
index 7a32e6328bc73..fb5a9c62b1317 100644
--- a/x-pack/plugins/ingest_pipelines/public/application/components/pipeline_processors_editor/components/test_pipeline/test_pipeline_flyout.container.tsx
+++ b/x-pack/plugins/ingest_pipelines/public/application/components/pipeline_processors_editor/components/test_pipeline/test_pipeline_flyout.container.tsx
@@ -19,6 +19,7 @@ import { TestPipelineFlyoutTab } from './test_pipeline_tabs';
 
 export interface Props {
   activeTab: TestPipelineFlyoutTab;
+  setActiveTab: (tab: TestPipelineFlyoutTab) => void;
   onClose: () => void;
   processors: DeserializeResult;
 }
@@ -35,6 +36,7 @@ export interface TestPipelineFlyoutForm {
 export const TestPipelineFlyout: React.FunctionComponent<Props> = ({
   onClose,
   activeTab,
+  setActiveTab,
   processors,
 }) => {
   const { services } = useKibana();
@@ -56,8 +58,6 @@ export const TestPipelineFlyout: React.FunctionComponent<Props> = ({
     },
   });
 
-  const [selectedTab, setSelectedTab] = useState<TestPipelineFlyoutTab>(activeTab);
-
   const [isRunningTest, setIsRunningTest] = useState<boolean>(false);
   const [testingError, setTestingError] = useState<any>(null);
   const [testOutput, setTestOutput] = useState<any>(undefined);
@@ -155,7 +155,7 @@ export const TestPipelineFlyout: React.FunctionComponent<Props> = ({
     });
 
     if (isSuccessful) {
-      setSelectedTab('output');
+      setActiveTab('output');
     }
   };
 
@@ -184,8 +184,8 @@ export const TestPipelineFlyout: React.FunctionComponent<Props> = ({
       testOutput={testOutput}
       form={form}
       validateAndTestPipeline={validateAndTestPipeline}
-      selectedTab={selectedTab}
-      setSelectedTab={setSelectedTab}
+      selectedTab={activeTab}
+      setSelectedTab={setActiveTab}
       testingError={testingError}
       onClose={onClose}
     />