diff --git a/Project.toml b/Project.toml
index 878f477..4ea64e2 100644
--- a/Project.toml
+++ b/Project.toml
@@ -5,7 +5,6 @@ version = "0.2.1"
 
 [deps]
 FunctionWrappers = "069b7b12-0de2-55c6-9aab-29f3d0a68a2e"
-Graphs = "86223c79-3864-5bf0-83f7-82e725a168b6"
 ITensorMPS = "0d1a4710-d33b-49a5-8f18-73bdf49b47e2"
 ITensors = "9136182c-28ba-11e9-034c-db9fb085ebd5"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
@@ -17,7 +16,6 @@ TimerOutputs = "a759f4b9-e2f1-59dc-863e-4aeb61b1ea8f"
 
 [compat]
 FunctionWrappers = "1.1.3"
-Graphs = "1.14.0"
 ITensorMPS = "0.1, 0.2, 0.3"
 ITensors = "0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9"
 LinearAlgebra = "1.9, 1.10"
@@ -27,9 +25,3 @@ SparseArrays = "1.9, 1.10"
 ThreadsX = "0.1.12"
 TimerOutputs = "0.5"
 julia = "1.10"
-
-[extras]
-Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
-
-[targets]
-test = ["Test"]
diff --git a/test/Project.toml b/test/Project.toml
new file mode 100644
index 0000000..ec6a311
--- /dev/null
+++ b/test/Project.toml
@@ -0,0 +1,14 @@
+[deps]
+Graphs = "86223c79-3864-5bf0-83f7-82e725a168b6"
+ITensorMPS = "0d1a4710-d33b-49a5-8f18-73bdf49b47e2"
+ITensors = "9136182c-28ba-11e9-034c-db9fb085ebd5"
+LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
+Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
+Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
+
+[compat]
+Graphs = "1.14.0"
+ITensorMPS = "0.1, 0.2, 0.3"
+ITensors = "0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9"
+LinearAlgebra = "1.9, 1.10"
+Random = "1"
diff --git a/test/large-graph-tests.jl b/test/large-graph-tests.jl
index a2d55a9..d752f46 100644
--- a/test/large-graph-tests.jl
+++ b/test/large-graph-tests.jl
@@ -11,7 +11,7 @@ using ITensorMPOConstruction:
 
 using Random
 using Test
-using Graphs # TODO: See if we can remove this dependency, since only used in testing.
+using Graphs
 
 function test_get_connected_components(nl::Int, nr::Int, max_edges_from_left::Int)
   g = BipartiteGraph{Int,Int,Float64}(