diff --git a/smt_design_space/test_design_space.py b/smt_design_space/test_design_space.py
index 625f182..4ceb488 100644
--- a/smt_design_space/test_design_space.py
+++ b/smt_design_space/test_design_space.py
@@ -7,7 +7,7 @@
 import unittest
 
 import numpy as np
-from SMTDesignSpace import design_space as ds
+from smt_design_space import design_space as ds
 
 from smt.sampling_methods import LHS
 from smt_design_space.design_space import (
@@ -690,7 +690,7 @@ def test_restrictive_value_constraint_categorical(self):
     )
     def test_adsg_to_legacy(self):
         from adsg_core import BasicADSG, NamedNode, DesignVariableNode
-        from smt_design_space.design_space import ensure_design_space
+        from smt_design_space   .design_space import ensure_design_space
         from adsg_core import GraphProcessor
 
         # Create the ADSG