diff --git a/main.py b/main.py
index 912bcc7..8f33b1c 100644
--- a/main.py
+++ b/main.py
@@ -32,7 +32,8 @@ if __name__ == '__main__':
     train['Y'] = train.apply(lambda x: (float(x.Beginning) + float(x.End))/2, axis=1)
     train = train.drop(columns=['Beginning', 'End', 'Title', 'Source'])
 
-    model = Pipeline([TfidfVectorizer(), LinearRegression()])
+    estimators = [('tfidf', TfidfVectorizer()), ('linearRegression', LinearRegression())]
+    model = Pipeline(estimators)
     model.fit(train.X, train.Y)
 
     # dev-0