lib/parsi-localize.rb in parsi-localize-0.3 vs lib/parsi-localize.rb in parsi-localize-0.3.1

@@ -14,13 +14,13 @@
         format = I18n.t("date.formats.#{format}") if object.is_a? Date
         format = I18n.t("time.formats.#{format}") if object.is_a? Time
       end
 
       if locale == :fa
-        if object.respond_to?(:with_parsi_digits)
-          object.with_parsi_digits
-        elsif [Date, DateTime, Time, Parsi::Date, Parsi::DateTime].include? object.class
+        if object.respond_to?(:to_jalali)
           object.to_jalali.strftime(format).with_parsi_digits
+        elsif object.respond_to?(:with_parsi_digits)
+          object.with_parsi_digits
         else
           config.backend.localize locale, object, format, options
         end
       else
         config.backend.localize locale, object, format, options