diff --git a/IUM_05-model.py b/IUM_05-model.py
index 89e1990..0f8a5a6 100644
--- a/IUM_05-model.py
+++ b/IUM_05-model.py
@@ -23,6 +23,6 @@ model.compile(optimizer='adam',
               loss='binary_crossentropy',
               metrics=['accuracy'])
 
-model.fit(X_train_pad, y_train, epochs=sys.argv[1], batch_size=sys.argv[2], validation_split=0.1)
+model.fit(X_train_pad, y_train, epochs=int(sys.argv[1]), batch_size=int(sys.argv[2]), validation_split=0.1)
 
 model.save('beer_review_sentiment_model.h5')