diff --git a/src/ITensorParallel.jl b/src/ITensorParallel.jl
index 574bdd3..bad9968 100644
--- a/src/ITensorParallel.jl
+++ b/src/ITensorParallel.jl
@@ -8,10 +8,12 @@ using MPI
 using ITensors
 using ITensors.NDTensors
 
-using ITensors: AbstractSum, Algorithm, @Algorithm_str
+using ITensors: Algorithm, @Algorithm_str
+using ITensors.ITensorMPS: AbstractSum
 
 import Base: eltype, length, size
-import ITensors:
+import ITensors: product
+import ITensors.ITensorMPS:
   disk,
   linkind,
   lproj,
@@ -19,15 +21,14 @@ import ITensors:
   nsite,
   orthogonalize!,
   position!,
-  product,
   replaceind!,
   replacebond!,
   rproj,
   set_terms,
   terms
-include(joinpath("partition", "partition.jl"))
-include(joinpath("partition", "partition_sum_split.jl"))
-include(joinpath("partition", "partition_chain_split.jl"))
+include("partition/partition.jl")
+include("partition/partition_sum_split.jl")
+include("partition/partition_chain_split.jl")
 include("force_gc.jl")
 include("foldssum.jl")
 include("distributedsum.jl")
diff --git a/test/Project.toml b/test/Project.toml
index 4c67d29..f645986 100644
--- a/test/Project.toml
+++ b/test/Project.toml
@@ -2,6 +2,7 @@
 ArgParse = "c7e460c6-2fb9-53a9-8c5b-16f535851c63"
 Compat = "34da2185-b29b-5c13-b0c7-acf172513d20"
 Distributed = "8ba89e20-285c-5b6f-9357-94700520ee1b"
+ITensorParallel = "0fccfcd2-f061-4985-9740-339d3f86bfce"
 ITensors = "9136182c-28ba-11e9-034c-db9fb085ebd5"
 MPI = "da04e1cc-30fd-572f-bb4f-1f8673147195"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"