diff --git a/lib/neo4jr/path_rater.rb b/lib/neo4jr/path_rater.rb
index 0c63c39..bc3f4b3 100644
--- a/lib/neo4jr/path_rater.rb
+++ b/lib/neo4jr/path_rater.rb
@@ -4,7 +4,7 @@ def get_cost path, start_cost, cost_evaluator, cost_accumulator
       cost = start_cost
       previous = nil
       path.each do |r|
-        cost = cost_accumulator.addCosts(cost, cost_evaluator.getCost(r, r.getEndNode == previous)) if r.kind_of? org.neo4j.api.core.Relationship
+        cost = cost_accumulator.addCosts(cost, cost_evaluator.getCost(r, r.getEndNode == previous)) if r.kind_of? org.neo4j.graphdb.Relationship
         previous = r
       end
       cost