diff --git a/correction.py b/correction.py
index c4e67b1..793bf35 100644
--- a/correction.py
+++ b/correction.py
@@ -158,7 +158,7 @@ if PROXY_URL:
 
 MODEL_ID_pro = "gemini-3.1-pro-preview"
 MODEL_ID_flash = "gemini-3-flash-preview"
-api_key="REMOVED_API_KEY"
+api_key = os.environ["GEMINI_API_KEY"]
 
 from pydantic import BaseModel, Field, TypeAdapter
 from typing import List, Optional, Tuple
diff --git a/gemini_for_labels.py b/gemini_for_labels.py
index ee048de..c87895a 100644
--- a/gemini_for_labels.py
+++ b/gemini_for_labels.py
@@ -14,7 +14,7 @@ from collections import defaultdict
 from concurrent.futures import ThreadPoolExecutor
 
 MODEL_ID = "gemini-3-flash-preview"
-api_key="REMOVED_API_KEY"
+api_key = os.environ["GEMINI_API_KEY"]
 
 my_prompt = """I'm giving you an image of the left columns of a written exam.
 Students answer several exercises, which can have several questions.