diff --git a/tests/problems/cross_modality/test_translation_problem.py b/tests/problems/cross_modality/test_translation_problem.py
index 81f2232d..ccd5de25 100644
--- a/tests/problems/cross_modality/test_translation_problem.py
+++ b/tests/problems/cross_modality/test_translation_problem.py
@@ -4,7 +4,6 @@
 import pytest
 
 import numpy as np
-from ott.geometry import epsilon_scheduler
 
 from anndata import AnnData
 
diff --git a/tests/problems/generic/test_fgw_problem.py b/tests/problems/generic/test_fgw_problem.py
index df1bb634..4167edf6 100644
--- a/tests/problems/generic/test_fgw_problem.py
+++ b/tests/problems/generic/test_fgw_problem.py
@@ -4,7 +4,6 @@
 
 import numpy as np
 import pandas as pd
-from ott.geometry import epsilon_scheduler
 from ott.geometry.costs import Cosine, Euclidean, PNormP, SqEuclidean, SqPNorm
 from ott.solvers.linear import acceleration
 
diff --git a/tests/problems/generic/test_gw_problem.py b/tests/problems/generic/test_gw_problem.py
index f92b0a67..5de5ea0b 100644
--- a/tests/problems/generic/test_gw_problem.py
+++ b/tests/problems/generic/test_gw_problem.py
@@ -4,7 +4,6 @@
 
 import numpy as np
 import pandas as pd
-from ott.geometry import epsilon_scheduler
 from ott.geometry.costs import Cosine, Euclidean, PNormP, SqEuclidean, SqPNorm
 from ott.solvers.linear import acceleration
 
diff --git a/tests/problems/generic/test_sinkhorn_problem.py b/tests/problems/generic/test_sinkhorn_problem.py
index 9892ee78..ef7c56f6 100644
--- a/tests/problems/generic/test_sinkhorn_problem.py
+++ b/tests/problems/generic/test_sinkhorn_problem.py
@@ -4,7 +4,6 @@
 
 import numpy as np
 import pandas as pd
-from ott.geometry import epsilon_scheduler
 from ott.geometry.costs import Cosine, Euclidean, PNormP, SqEuclidean, SqPNorm
 from ott.solvers.linear import acceleration
 
diff --git a/tests/problems/space/test_alignment_problem.py b/tests/problems/space/test_alignment_problem.py
index 901f46d7..b3e21cee 100644
--- a/tests/problems/space/test_alignment_problem.py
+++ b/tests/problems/space/test_alignment_problem.py
@@ -6,7 +6,6 @@
 import numpy as np
 import pandas as pd
 import scipy.sparse as sp
-from ott.geometry import epsilon_scheduler
 
 import scanpy as sc
 from anndata import AnnData
diff --git a/tests/problems/space/test_mapping_problem.py b/tests/problems/space/test_mapping_problem.py
index 7d652029..63ffdb17 100644
--- a/tests/problems/space/test_mapping_problem.py
+++ b/tests/problems/space/test_mapping_problem.py
@@ -7,7 +7,6 @@
 import numpy as np
 import pandas as pd
 import scipy.sparse as sp
-from ott.geometry import epsilon_scheduler
 from ott.solvers.linear.sinkhorn import SinkhornOutput
 from ott.solvers.quadratic.gromov_wasserstein import GWOutput
 
diff --git a/tests/problems/spatio_temporal/test_spatio_temporal_problem.py b/tests/problems/spatio_temporal/test_spatio_temporal_problem.py
index 3632d76b..4dd31fa5 100644
--- a/tests/problems/spatio_temporal/test_spatio_temporal_problem.py
+++ b/tests/problems/spatio_temporal/test_spatio_temporal_problem.py
@@ -4,7 +4,6 @@
 
 import numpy as np
 import pandas as pd
-from ott.geometry import epsilon_scheduler
 
 from anndata import AnnData
 
diff --git a/tests/problems/time/test_temporal_problem.py b/tests/problems/time/test_temporal_problem.py
index d3780a8e..c5ec8e08 100644
--- a/tests/problems/time/test_temporal_problem.py
+++ b/tests/problems/time/test_temporal_problem.py
@@ -6,7 +6,7 @@
 import numpy as np
 import pandas as pd
 import scipy.sparse as sp
-from ott.geometry import costs, epsilon_scheduler
+from ott.geometry import costs
 from scipy.sparse import csr_matrix
 
 import scanpy as sc