1 files changed, 11 insertions, 4 deletions
diff --git a/web/src/js/components/FlowView/Messages.jsx b/web/src/js/components/FlowView/Messages.jsx
index 27e18c05..8cb918a8 100644
--- a/web/src/js/components/FlowView/Messages.jsx
+++ b/web/src/js/components/FlowView/Messages.jsx
@@ -7,6 +7,8 @@ import ContentView from '../ContentView'
 import ValueEditor from '../ValueEditor'
 import Headers from './Headers'
 import * as flowActions  from '../../ducks/flows'
+import FlowEditorButton from './FlowEditorButton.jsx'
+
 
 class RequestLine extends Component {
 
@@ -77,21 +79,23 @@ class ResponseLine extends Component {
 }
 
 export class Request extends Component {
-
-    render() {
+   render() {
         const { flow, updateFlow } = this.props
+        let onContentChange = content => flowActions.updateContent(this.props.flow, content, "request")
 
         return (
             <section className="request">
+                <FlowEditorButton onContentChange={onContentChange}/>
                 <RequestLine ref="requestLine" flow={flow} updateFlow={updateFlow} />
                 <Headers
                     ref="headers"
                     message={flow.request}
                     onChange={headers => updateFlow({ request: { headers } })}
                 />
+
                 <hr/>
                 <ContentView flow={flow}
-                             onContentChange={content => flowActions.updateContent(this.props.flow, content, "request") }
+                             onContentChange={onContentChange}
                              message={flow.request}
                 />
             </section>
@@ -120,11 +124,14 @@ export class Request extends Component {
 
 export class Response extends Component {
 
+
     render() {
         const { flow, updateFlow } = this.props
+        let onContentChange = content => flowActions.updateContent(this.props.flow, content, "response")
 
         return (
             <section className="response">
+                <FlowEditorButton onContentChange={onContentChange}/>
                 <ResponseLine ref="responseLine" flow={flow} updateFlow={updateFlow} />
                 <Headers
                     ref="headers"
@@ -133,7 +140,7 @@ export class Response extends Component {
                 />
                 <hr/>
                 <ContentView flow={flow}
-                             onContentChange={content => flowActions.updateContent(this.props.flow, content, "response") }
+                             onContentChange={onContentChange}
                              message={flow.response}
                 />
             </section>