diff --git a/jedi/evaluate/context/__init__.py b/jedi/evaluate/context/__init__.py
index b227a1d3..0047f15c 100644
--- a/jedi/evaluate/context/__init__.py
+++ b/jedi/evaluate/context/__init__.py
@@ -1,3 +1,5 @@
-from jedi.evaluate.context.base import *
+from jedi.evaluate.context.base import Context, iterate_contexts, \
+    TreeContext, ContextualizedName, ContextualizedNode, ContextSet, \
+    NO_CONTEXTS, iterator_to_context_set
 from jedi.evaluate.context.lazy import AbstractLazyContext, LazyKnownContext, \
     LazyKnownContexts, LazyTreeContext, LazyUnknownContext, get_merged_lazy_context