diff --git a/Topological Sort/TopologicalSort1.swift b/Topological Sort/TopologicalSort1.swift
index 2ad9b51a0..05f1ab887 100644
--- a/Topological Sort/TopologicalSort1.swift	
+++ b/Topological Sort/TopologicalSort1.swift	
@@ -1,6 +1,7 @@
 extension Graph {
   private func depthFirstSearch(_ source: Node, visited: inout [Node : Bool]) -> [Node] {
     var result = [Node]()
+    visited[source] = true
 
     if let adjacencyList = adjacencyList(forNode: source) {
       for nodeInAdjacencyList in adjacencyList {
@@ -10,7 +11,6 @@ extension Graph {
       }
     }
 
-    visited[source] = true
     return [source] + result
   }