diff --git a/images/odoo/bin/remove-ssh-key b/images/odoo/bin/remove-ssh-key
index c4d3cd5d..63aff554 100755
--- a/images/odoo/bin/remove-ssh-key
+++ b/images/odoo/bin/remove-ssh-key
@@ -3,12 +3,11 @@ set -e
 
 if [[ -n "$GIT_SSH_PRIVATE_KEY" ]]; then
     key_filename="${SSH_ID_ALGORITHM:=id_ed25519}"
-
     log-entrypoint 'Remove SSH key from env var.'
-    if grep -u "$(id -u)" ssh-agent > /dev/null; then
+    if [[ -n "$SSH_AUTH_SOCK" ]] || [[ -n "$SSH_AGENT_PID" ]]; then
         if [[ -f "$HOME/.ssh/$key_filename" ]]; then
             ssh-add -d "$HOME/.ssh/$key_filename" 2>/dev/null || true
         fi
     fi
     rm -f "$HOME/.ssh/$key_filename"
-fi
+fi
\ No newline at end of file
diff --git a/task b/task
index e882f989..bef18d89 100755
--- a/task
+++ b/task
@@ -1544,7 +1544,7 @@ def migrate(cr, version):
     _logger.info("Updated %s partners", len(partners))
 EOF
         sed -i "/\"version\":/c\    \"version\": \"${ODOO_VERSION}.2.0.0\"," oca/partner-contact/partner_firstname/__manifest__.py
-        exec odoo update-module partner_firstname 2>&1 | grep "module partner_firstname: Running migration"
+        odoo update-module partner_firstname 2>&1 | grep "module partner_firstname: Running migration" || true
     fi
 }