lib/flexirest/request.rb in flexirest-1.7.3 vs lib/flexirest/request.rb in flexirest-1.7.4

@@ -237,16 +237,17 @@
           end
 
           if object_is_class? && @object.record_response?
             @object.record_response(self.url, response_env)
           end
-          if object_is_class?
-            callback_result = @object.send(:_callback_request, :after, @method[:name], response_env)
-          else
-            callback_result = @object.class.send(:_callback_request, :after, @method[:name], response_env)
-          end
 
-          if callback_result == :retry
+          begin
+            if object_is_class?
+              callback_result = @object.send(:_callback_request, :after, @method[:name], response_env)
+            else
+              callback_result = @object.class.send(:_callback_request, :after, @method[:name], response_env)
+            end
+          rescue Flexirest::CallbackRetryRequestException
             if self.retrying != true
               self.retrying = true
               return call()
             end
           end