lib/neo4j/active_node/property.rb in neo4j-3.0.0.rc.2 vs lib/neo4j/active_node/property.rb in neo4j-3.0.0.rc.3

@@ -1,18 +1,22 @@
 module Neo4j::ActiveNode
   module Property
     extend ActiveSupport::Concern
     include Neo4j::Shared::Property
 
+    def initialize(attributes={}, options={})
+      super(attributes, options)
+
+      send_props(@relationship_props) if persisted? and not @relationship_props.nil?
+    end
+
     module ClassMethods
 
       # Extracts keys from attributes hash which are relationships of the model
       # TODO: Validate separately that relationships are getting the right values?  Perhaps also store the values and persist relationships on save?
       def extract_association_attributes!(attributes)
-        attributes.keys.inject({}) do |association_props, key|
+        attributes.keys.each_with_object({}) do |key, association_props|
           association_props[key] = attributes.delete(key) if self.has_association?(key)
-
-          association_props
         end
       end
     end
   end
 end