Update Content Type Invoker

Apatis · Mar 24, 2018 · 289941f · 289941f
1 parent 767ddb1
commit 289941f
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/NotAllowedHandler.php b/src/NotAllowedHandler.php
@@ -245,9 +245,11 @@ public function __invoke(
     ): ResponseInterface {
         $contentType = $this->getContentType();
         // if content type has not been set
-        // get it from Request
+        // get it from Request / Response
         if (!$contentType) {
-            $contentType = $request->getHeaderLine('Content-Type')?: static::DEFAULT_CONTENT_TYPE;
+            $contentType = $response->getHeaderLine('Content-Type')?: (
+                $request->getHeaderLine('Content-Type')?: static::DEFAULT_CONTENT_TYPE
+            );
             $this->setContentType($contentType);
         }