diff --git a/docs/concepts/test_data_generation/rag.md b/docs/concepts/test_data_generation/rag.md
index fa655ad46..2b93e72ef 100644
--- a/docs/concepts/test_data_generation/rag.md
+++ b/docs/concepts/test_data_generation/rag.md
@@ -170,7 +170,7 @@ You can write your own [custom relationship builder]() to establish the relation
 
 ```python
 from ragas.testset.graph import KnowledgeGraph
-from ragas.testset.transforms.relationship_builders.cosine import JaccardSimilarityBuilder
+from ragas.testset.transforms.relationship_builders.traditional import JaccardSimilarityBuilder
 
 kg = KnowledgeGraph(nodes=sample_nodes)
 rel_builder = JaccardSimilarityBuilder(property_name="entities", key_name="PER", new_property_name="entity_jaccard_similarity")
@@ -287,4 +287,4 @@ class EntityQuerySynthesizer(QuerySynthesizer):
         """
 
         return SingleTurnSample(user_input=query, reference_contexs=contexts, reference=reference)
-```
\ No newline at end of file
+```