diff --git a/docs/assets/use_cases/node_representation_learning/context_proba.png b/docs/assets/use_cases/node_representation_learning/context_proba.png
index 571dde994..caddfc1bc 100644
Binary files a/docs/assets/use_cases/node_representation_learning/context_proba.png and b/docs/assets/use_cases/node_representation_learning/context_proba.png differ
diff --git a/docs/assets/use_cases/node_representation_learning/sage_layer_eqn.png b/docs/assets/use_cases/node_representation_learning/sage_layer_eqn.png
index 26e60110a..c92c59935 100644
Binary files a/docs/assets/use_cases/node_representation_learning/sage_layer_eqn.png and b/docs/assets/use_cases/node_representation_learning/sage_layer_eqn.png differ