diff --git a/nni/nas/__init__.py b/nni/nas/__init__.py
index 588bb70f3a..358e881832 100644
--- a/nni/nas/__init__.py
+++ b/nni/nas/__init__.py
@@ -1,4 +1,5 @@
 # Copyright (c) Microsoft Corporation.
 # Licensed under the MIT license.
 
+from .space import *
 from .utils import *