diff --git a/files/tal/on_update.sh b/files/tal/on_update.sh
index 068b662364711855ac1e0e8c6a085896c7b4fb2b..4c364f7e69ef4e32e8e09174334a1bad7eba74e7 100644
--- a/files/tal/on_update.sh
+++ b/files/tal/on_update.sh
@@ -1,5 +1,11 @@
 #!/usr/bin/bash
 
+if [[ $# -gt 0 ]]
+then
+    /home/examadm/tal-cli/on_update.sh | tee /home/examadm/tal-cli/log
+    exit $?
+fi
+
 cd source
 
 python3 -m pip install --user -r scripts/requirements.txt
diff --git a/manifests/tal_cli.pp b/manifests/tal_cli.pp
index ed6610adbb3cd76602c8fb15e87b7f8d27a67868..777a78d4cff8c04d68965237f24ab1ab753cf156 100644
--- a/manifests/tal_cli.pp
+++ b/manifests/tal_cli.pp
@@ -74,7 +74,7 @@ class aes::tal_cli (
     path        => '/bin:/usr/bin',
     environment => ['HOME=/home/examadm'],
     creates     => '/home/examadm/tal-cli/source/tal',
-    command     => '/home/examadm/tal-cli/on_update.sh | tee /home/examadm/tal-cli/log.txt',
+    command     => '/home/examadm/tal-cli/on_update.sh --log',
     require     => File['/home/examadm/tal-cli/on_update.sh'],
   }