diff --git a/environments/dataset_environment/dataset_env.py b/environments/dataset_environment/dataset_env.py
index ecfc9ec2..c29d0d50 100644
--- a/environments/dataset_environment/dataset_env.py
+++ b/environments/dataset_environment/dataset_env.py
@@ -241,7 +241,7 @@ class DatasetEnv(BaseEnv):
             top_p=self.config.top_p,
         )
 
-        print(f"Completions: {completions}")
+        logger.debug(f"Completions: {completions}")
         # Process completions
         trajectories = []
         for completion in completions.choices:
diff --git a/environments/infinimath/infinimath_env.py b/environments/infinimath/infinimath_env.py
index 6a4d2eb7..f7472522 100644
--- a/environments/infinimath/infinimath_env.py
+++ b/environments/infinimath/infinimath_env.py
@@ -539,7 +539,7 @@ class InfiniteMathEnv(BaseEnv):
                 if hasattr(completion, "text")
                 else completion.message.content
             )
-            print("model_answer", model_answer)
+            logger.debug(f"model_answer: {model_answer}")
 
             full_messages = [
                 {"role": "system", "content": self.system_prompt},