diff --git a/graphs/basic_graphs.py b/graphs/basic_graphs.py
index 8cdde6abc819..e0a7234620c6 100644
--- a/graphs/basic_graphs.py
+++ b/graphs/basic_graphs.py
@@ -220,6 +220,7 @@ def prim(G, s):
                 if v[1] < dist.get(v[0], 100000):
                     dist[v[0]] = v[1]
                     path[v[0]] = u
+    return dist
 
 
 """