diff --git a/docs/source/examples/gatne_examples_code.rst b/docs/source/examples/gatne_examples_code.rst
index aea3404e1a15b4ab11d86cca977dc62d4d7d3644..326d07b2b43465227a148b6e3fb7b6a5201cf74c 100644
--- a/docs/source/examples/gatne_examples_code.rst
+++ b/docs/source/examples/gatne_examples_code.rst
@@ -1,8 +1,8 @@
 View the Code
 =============
 
-examples/dgi/train.py
+examples/GATNE/main.py
 
-.. literalinclude:: ../../../examples/dgi/train.py
+.. literalinclude:: ../../../examples/GATNE/main.py
     :language: python
     :linenos: