lib/neo4j/active_node/labels.rb in neo4j-3.0.0.alpha.9 vs lib/neo4j/active_node/labels.rb in neo4j-3.0.0.alpha.10

@@ -11,23 +11,23 @@
       class RecordNotFound < StandardError; end
 
       # @return the labels
       # @see Neo4j-core
       def labels
-        @_persisted_node.labels
+        @_persisted_obj.labels
       end
 
       # adds one or more labels
       # @see Neo4j-core
       def add_label(*label)
-        @_persisted_node.add_label(*label)
+        @_persisted_obj.add_label(*label)
       end
 
       # Removes one or more labels
       # Be careful, don't remove the label representing the Ruby class.
       # @see Neo4j-core
       def remove_label(*label)
-        @_persisted_node.remove_label(*label)
+        @_persisted_obj.remove_label(*label)
       end
 
       def self.included(klass)
         add_wrapped_class(klass)
       end