diff --git a/{{cookiecutter.project_name}}/{% if cookiecutter.__ci=='gitlab' %}.gitlab-ci.yml{% endif %} b/{{cookiecutter.project_name}}/{% if cookiecutter.__ci=='gitlab' %}.gitlab-ci.yml{% endif %}
index fff26c35..3d6fea47 100644
--- a/{{cookiecutter.project_name}}/{% if cookiecutter.__ci=='gitlab' %}.gitlab-ci.yml{% endif %}	
+++ b/{{cookiecutter.project_name}}/{% if cookiecutter.__ci=='gitlab' %}.gitlab-ci.yml{% endif %}	
@@ -164,6 +164,9 @@ deploy_staging:
   rules:
     - if:
         $CI_COMMIT_BRANCH == $CI_DEFAULT_BRANCH && $CI_PIPELINE_SOURCE == "push"
+  id_tokens:
+    PYPI_ID_TOKEN:
+      aud: testpypi
   variables:
     TWINE_REPOSITORY: testpypi
     TWINE_USERNAME: __token__
@@ -173,6 +176,9 @@ deploy_production:
   extends: .deploy
   only:
     - tags
+  id_tokens:
+    PYPI_ID_TOKEN:
+      aud: pypi
   variables:
     TWINE_REPOSITORY: pypi
     TWINE_USERNAME: __token__