diff --git a/DGSD/main.py b/DGSD/main.py
index fdb6782..2cbd9ad 100644
--- a/DGSD/main.py
+++ b/DGSD/main.py
@@ -21,6 +21,8 @@ def get_descriptor(self, graph, bins = 50, workers=1):
             batches = np.array_split(nodes, workers)
         emb = p.starmap(self.Generate_Embeddings, zip(batches, repeat(bins)))
         embeddings = np.sum(np.array(emb),axis = 0)
+        p.close()
+        p.join()
         return embeddings
 
     def Generate_Embeddings(self, batch, nbins):