lib/sequel/plugins/bitemporal.rb in sequel_bitemporal-0.6.10 vs lib/sequel/plugins/bitemporal.rb in sequel_bitemporal-0.6.11

@@ -237,11 +237,13 @@
         end
 
         def restore(attrs={})
           return false unless deleted?
           last_version_attributes = if last_version
-            last_version.values.reject do |column, _|
-              excluded_columns.include? column
+            last_version.columns.each_with_object({}) do |column, hash|
+              unless excluded_columns.include? column
+                hash[column] = last_version.send column
+              end
             end
           else
             {}
           end
           update_attributes last_version_attributes.merge attrs