diff --git a/app/controllers/admin/price_documents_controller.rb b/app/controllers/admin/price_documents_controller.rb
index 6966596..2eb5ca2 100644
--- a/app/controllers/admin/price_documents_controller.rb
+++ b/app/controllers/admin/price_documents_controller.rb
@@ -340,14 +340,13 @@ class Admin::PriceDocumentsController < ApplicationController
 
     @avoir = true if @price_document.label == "Avoir"
     if @price_document.update_attributes(params.require(:price_document).permit!)
-      
       if params[:public_edit].to_s == "true"
         
         @price_document.reset_for_update
         
         #@price_document.unarchive_now
         #@price_document.archive_now
-        if @price_document.cc_label = "Facture achat"
+        if @price_document.cc_label == "Facture achat"
           redirect_to [:admin, @price_document]
         elsif @price_document.doc_ref_id
           redirect_to [:admin, PriceDocument.find(@price_document.doc_ref_id)]
@@ -356,6 +355,11 @@ class Admin::PriceDocumentsController < ApplicationController
         else
           redirect_to [:admin, @price_document]
         end
+
+      elsif @price_document.doc_ref_id
+        redirect_to [:admin, PriceDocument.find(@price_document.doc_ref_id)]
+      elsif @price_document.ref_element
+        redirect_to [:admin, @price_document.ref_element]
       else
         redirect_to [:admin, @price_document]
       end