diff --git a/risk/__init__.py b/risk/__init__.py
index 88a4af9..2134f25 100644
--- a/risk/__init__.py
+++ b/risk/__init__.py
@@ -48,4 +48,4 @@ mydf.risk.evaluate(pop=pop)
     - Evaluation of how sparse attributes are (the ratio of non-null over rows)
     - Have a smart way to drop attributes (based on the above in random policy search)
 """
-from risk import *
+from risk import risk