diff --git a/smt/design_space/__init__.py b/smt/design_space/__init__.py
index 77a3e8fa3..a88f687dc 100644
--- a/smt/design_space/__init__.py
+++ b/smt/design_space/__init__.py
@@ -5,6 +5,7 @@
     IntegerVariable,
     OrdinalVariable,
     DesignSpace,
+    DesignVariable,
     ensure_design_space,
 )
 
@@ -15,6 +16,6 @@
     "OrdinalVariable",
     "CategoricalVariable",
     "DesignSpace",
+    "DesignVariable",
     "ensure_design_space",
-    "HAS_DESIGN_SPACE_EXT",
 ]