diff --git a/lib/OptimalBranchingMIS/Project.toml b/lib/OptimalBranchingMIS/Project.toml
index 1e901bb..b85c882 100644
--- a/lib/OptimalBranchingMIS/Project.toml
+++ b/lib/OptimalBranchingMIS/Project.toml
@@ -12,7 +12,7 @@ OptimalBranchingCore = "c76e7b22-e1d2-40e8-b0f1-f659837787b8"
 [compat]
 Combinatorics = "1"
 EliminateGraphs = "0.2"
-GenericTensorNetworks = "2"
+GenericTensorNetworks = "3"
 OptimalBranchingCore = "0.1.0"
 julia = "1.10"
 
diff --git a/lib/OptimalBranchingMIS/src/OptimalBranchingMIS.jl b/lib/OptimalBranchingMIS/src/OptimalBranchingMIS.jl
index fd1dae1..4f50668 100644
--- a/lib/OptimalBranchingMIS/src/OptimalBranchingMIS.jl
+++ b/lib/OptimalBranchingMIS/src/OptimalBranchingMIS.jl
@@ -1,6 +1,7 @@
 module OptimalBranchingMIS
 
 using OptimalBranchingCore
+using OptimalBranchingCore: AbstractProblem
 using OptimalBranchingCore.BitBasis
 using Combinatorics
 using EliminateGraphs, EliminateGraphs.Graphs