diff --git a/biblioteki_ml.py b/biblioteki_ml.py
index 6f8e0a0..4f93aad 100644
--- a/biblioteki_ml.py
+++ b/biblioteki_ml.py
@@ -8,7 +8,7 @@ import pandas as pd
 
 # Parametry z konsoli
 try:
-    epochs = sys.argv[1]
+    epochs = int(sys.argv[1])
 except:
     print('No epoch number passed. Defaulting to 100')
     epochs = 100