diff --git a/lib/neography/node.rb b/lib/neography/node.rb index f6f762c..0f8ecec 100644 --- a/lib/neography/node.rb +++ b/lib/neography/node.rb @@ -13,7 +13,7 @@ def create(*args) # the arguments can be an hash of properties to set or a rest instance props = (args[0].respond_to?(:each_pair) && args[0]) || args[1] db = (args[0].is_a?(Neography::Rest) && args[0]) || args[1] || Neography::Rest.new - node = self.class.new(db.create_node(props)) + node = self.new(db.create_node(props)) node.neo_server = db node end @@ -25,7 +25,7 @@ def load(*args) # a db instance can be given, it is the first argument or the second db = (args[0].is_a?(Neography::Rest) && args[0]) || args[1] || Neography::Rest.new node = db.get_node(node) - node = self.class.new(node) unless node.nil? + node = self.new(node) unless node.nil? node.neo_server = db unless node.nil? node end