lib/rflow/configuration/uuid_keyed.rb in rflow-0.0.5 vs lib/rflow/configuration/uuid_keyed.rb in rflow-1.0.0a1

@@ -3,15 +3,15 @@
 class RFlow
   class Configuration
     module UUIDKeyed
       def self.included(base)
         base.class_eval do
-          set_primary_key 'uuid'
+          self.primary_key = 'uuid'
           before_create :generate_uuid
-          
+
           def generate_uuid
             self.uuid = UUIDTools::UUID.random_create.to_s
-          end 
+          end
         end
       end
     end
   end
 end