diff --git a/train.py b/train.py
index a610090..f80a4eb 100644
--- a/train.py
+++ b/train.py
@@ -39,7 +39,7 @@ def my_main(EPOCHS, _run):
     norm_games_model.save('test')
 
 exint.run()
-exint.add_artifact(filename="test/saved_model.pb")
-exint.add_artifact(filename="./ium_z434686/metrics.csv")
-exint.add_artifact(filename="./ium_z434686/metrics.png")
+exint.add_artifact('test/saved_model.pb', 'test')
+exint.add_artifact('ium_z434686/metrics.csv', 'csv')
+exint.add_artifact('ium_z434686/metrics.png', 'png')
 exint.add_source_file('./ium_z434686/train.py')