diff --git a/tests/test_graph_generation.py b/tests/test_graph_generation.py
index d1d6d69..604afb6 100644
--- a/tests/test_graph_generation.py
+++ b/tests/test_graph_generation.py
@@ -1,6 +1,5 @@
 import json
 import unittest
-from random import sample
 from typing import Any, Dict, List
 
 from pargraph import GraphEngine, delayed, graph