diff --git a/model.py b/model.py
index 1737a5e..9c57176 100644
--- a/model.py
+++ b/model.py
@@ -35,6 +35,6 @@ pipeline = Pipeline(steps=[
 
 pipeline['model'].compile(optimizer='adam', loss='mse', metrics=['mae'])
 
-pipeline.fit(X_train.iloc[:, 1:], y_train,  model__epochs=int(sys.argv[1]), model__validation_split=0.1)
+pipeline.fit(X_train.iloc[:, 1:], y_train.iloc[:, 1:],  model__epochs=int(sys.argv[1]), model__validation_split=0.1)
 
 pipeline['model'].save('powerlifting_model.h5')