diff --git a/pygsp/graphs/learngraph.py b/pygsp/graphs/learngraph.py
index 87ea8aec..246e020b 100644
--- a/pygsp/graphs/learngraph.py
+++ b/pygsp/graphs/learngraph.py
@@ -509,7 +509,7 @@ class LearnGraph(Graph):
     >>> 
     >>> # B) Learn the graph
     >>> param_opt = {'verbosity':0}
-    >>> Glearned = LearnGraph(X,k=k, param_opt=param_opt)
+    >>> Glearned = pg.graphs.LearnGraph(X,k=k, param_opt=param_opt)
     >>> # plot the learned graph
     >>> Glearned.coords = coords
     >>> 
diff --git a/pygsp/tests/test_learning.py b/pygsp/tests/test_learning.py
index b960d8a9..1d838e96 100644
--- a/pygsp/tests/test_learning.py
+++ b/pygsp/tests/test_learning.py
@@ -132,3 +132,6 @@ def test_classification_tikhonov(self):
 
 
 suite = unittest.TestLoader().loadTestsFromTestCase(TestCase)
+
+if __name__ == '__main__':
+    unittest.main()
\ No newline at end of file