diff --git a/model.py b/model.py
index f70bb98..9e18c49 100644
--- a/model.py
+++ b/model.py
@@ -11,7 +11,7 @@ import tensorflow as tf
 data = pd.read_csv('./data/train.csv')
 
 data = data[['Sex', 'Age', 'BodyweightKg', 'TotalKg']].dropna()
-data['Age'] = data['Age'].astype(int)
+data['Age'] = pd.to_numeric(data['Age'], errors='coerce')
 features = data[['Sex', 'Age', 'BodyweightKg']]
 target = data['TotalKg']