diff --git a/model.py b/model.py
index d761a2d..240a59a 100644
--- a/model.py
+++ b/model.py
@@ -35,8 +35,8 @@ pipeline = Pipeline(steps=[
 
 pipeline['model'].compile(optimizer='adam', loss='mse', metrics=['mae'])
 
-X_train_excluded = X_train.iloc[:, 1:]
-y_train_excluded = y_train.iloc[:, 1:]
+X_train_excluded = X_train.iloc[1:, :]
+y_train_excluded = y_train.iloc[1:, :]
 
 pipeline.fit(X_train_excluded, y_train_excluded, model__epochs=10, model__validation_split=0.1)