diff --git a/training.py b/training.py
index bed9999..ab07de1 100644
--- a/training.py
+++ b/training.py
@@ -51,4 +51,4 @@ exint.run()
 
 exint.add_artifact('test/saved_model.pb')
 
-exint.add_source_file('./ium_z434743/train.py')
+exint.add_source_file('./ium_z434743/training.py')