diff --git a/README.md b/README.md
index 061422be..a05de5fc 100644
--- a/README.md
+++ b/README.md
@@ -45,8 +45,7 @@ You can setup a full stack machine learning pipeline in a few lines of code:
 from sklearn.datasets import load_breast_cancer
 from sklearn.model_selection import KFold
 
-from photonai.base import Hyperpipe, PipelineElement
-from photonai.optimization import FloatRange, Categorical, IntegerRange
+from photonai import Hyperpipe, PipelineElement, FloatRange, Categorical, IntegerRange
 
 # DESIGN YOUR PIPELINE
 my_pipe = Hyperpipe('basic_svm_pipe',  # the name of your pipeline