diff --git a/pages/students/2016/jakub_maruniak/dp2021/annotation/train/train.sh b/pages/students/2016/jakub_maruniak/dp2021/annotation/train/train.sh
index a0d1c7cfc1..6c4f40d8ad 100644
--- a/pages/students/2016/jakub_maruniak/dp2021/annotation/train/train.sh
+++ b/pages/students/2016/jakub_maruniak/dp2021/annotation/train/train.sh
@@ -1,19 +1,21 @@
 set -e
 OUTDIR=build/train/output
 TRAINDIR=build/train
+# Delete old training results
+rm -r $TRAINDIR
 mkdir -p $TRAINDIR
 mkdir -p $OUTDIR
 mkdir -p dist
 # Delete old training results
 rm -rf $OUTDIR/*
 # Train dependency and POS
-spacy train sk $OUTDIR ./build/input/slovak-treebank ./build/input/ud-artificial-gapping  --n-iter 20 -p tagger,parser
+spacy train sk $OUTDIR ./build/input/slovak-treebank ./build/input/ud-artificial-gapping  --n-iter 15 -p tagger,parser
 rm -rf $TRAINDIR/posparser
 mv $OUTDIR/model-best $TRAINDIR/posparser
 # Train NER
 # python ./train.py -t ./train.json -o $TRAINDIR/nerposparser -n 10 -m $TRAINDIR/posparser/
-spacy train sk $TRAINDIR/nerposparser ./ner/train.json ./ner/eval.json --n-iter 20 -p ner
+spacy train sk $TRAINDIR/nerposparser ./ner/wikiart/wikitrain.json ./ner/wikiart/wikieval.json --n-iter 15 -p ner
 # Package model
 spacy package $TRAINDIR/nerposparser dist --meta-path ./meta.json --force
 cd dist/sk_sk1-0.2.0
-python ./setup.py sdist --dist-dir ../
+python ./setup.py sdist --dist-dir ../
\ No newline at end of file