diff --git a/tests/unit_tests/sampling/test_adaptive_kriging.py b/tests/unit_tests/sampling/test_adaptive_kriging.py
index 4fcf3b1e..e191feb3 100644
--- a/tests/unit_tests/sampling/test_adaptive_kriging.py
+++ b/tests/unit_tests/sampling/test_adaptive_kriging.py
@@ -1,6 +1,7 @@
 import pytest
 
-from UQpy import GaussianProcessRegression, RBF, LinearRegression
+from UQpy import GaussianProcessRegression, LinearRegression
+from UQpy.utilities.kernels.euclidean_kernels.RBF import RBF
 from UQpy.run_model.model_execution.PythonModel import PythonModel
 from UQpy.utilities.MinimizeOptimizer import MinimizeOptimizer
 
diff --git a/tests/unit_tests/sampling/test_refined_stratified.py b/tests/unit_tests/sampling/test_refined_stratified.py
index 13afea15..03e7921d 100644
--- a/tests/unit_tests/sampling/test_refined_stratified.py
+++ b/tests/unit_tests/sampling/test_refined_stratified.py
@@ -1,7 +1,8 @@
 import pytest
 from beartype.roar import BeartypeCallHintPepParamException
 
-from UQpy import GaussianProcessRegression, RBF, LinearRegression
+from UQpy import GaussianProcessRegression, LinearRegression
+from UQpy.utilities.kernels.euclidean_kernels.RBF import RBF
 from UQpy.run_model.model_execution.PythonModel import PythonModel
 from UQpy.utilities.MinimizeOptimizer import MinimizeOptimizer
 from UQpy.sampling.stratified_sampling.refinement.GradientEnhancedRefinement import GradientEnhancedRefinement