diff --git a/train.py b/train.py
index 2c82a6b..2ce55a7 100644
--- a/train.py
+++ b/train.py
@@ -37,8 +37,7 @@ def my_main(EPOCHS, _run):
     norm_games_model.fit(games_all, games_predict, epochs=EPOCHS)
 
     norm_games_model.save('test')
-    _run.add_artifact('test')
 
 exint.run()
-exint.add_artifact(filename="test")
+exint.add_artifact(filename="test/saved_model.pb")
 exint.add_source_file('./ium_z434686/train.py')