diff --git a/docs/Training-on-Amazon-Web-Service.md b/docs/Training-on-Amazon-Web-Service.md
index 132555fcce..52597ca144 100644
--- a/docs/Training-on-Amazon-Web-Service.md
+++ b/docs/Training-on-Amazon-Web-Service.md
@@ -194,7 +194,7 @@ linux executables which use visual observations.
 7. Test the instance setup from Python using:
 
     ```python
-    from unityagents import UnityEnvironment
+    from mlagents.envs import UnityEnvironment
 
     env = UnityEnvironment(<your_env>)
     ```
@@ -207,4 +207,4 @@ linux executables which use visual observations.
     //cd into your ml-agents/python folder
     chmod +x <your_env>.x86_64
     python learn.py <your_env> --train
-    ```
\ No newline at end of file
+    ```
diff --git a/docs/Training-on-Microsoft-Azure.md b/docs/Training-on-Microsoft-Azure.md
index 88c30f769b..6aa60099cc 100644
--- a/docs/Training-on-Microsoft-Azure.md
+++ b/docs/Training-on-Microsoft-Azure.md
@@ -50,7 +50,7 @@ To verify that all steps worked correctly:
 7. Test the instance setup from Python using:
 
 ```python
-from unityagents import UnityEnvironment
+from mlagents.envs import UnityEnvironment
 
 env = UnityEnvironment(<your_env>)
 ```
diff --git a/ml-agents/mlagents/trainers/bc/policy.py b/ml-agents/mlagents/trainers/bc/policy.py
index 3cbb460c9e..585d8eb6fe 100644
--- a/ml-agents/mlagents/trainers/bc/policy.py
+++ b/ml-agents/mlagents/trainers/bc/policy.py
@@ -4,7 +4,7 @@
 from mlagents.trainers.bc.models import BehavioralCloningModel
 from mlagents.trainers.policy import Policy
 
-logger = logging.getLogger("unityagents")
+logger = logging.getLogger("mlagents.trainers")
 
 
 class BCPolicy(Policy):
diff --git a/ml-agents/mlagents/trainers/policy.py b/ml-agents/mlagents/trainers/policy.py
index 3f3e7a5c6c..8578af7c45 100644
--- a/ml-agents/mlagents/trainers/policy.py
+++ b/ml-agents/mlagents/trainers/policy.py
@@ -4,7 +4,7 @@
 from mlagents.trainers import UnityException
 from mlagents.trainers.models import LearningModel
 
-logger = logging.getLogger("unityagents")
+logger = logging.getLogger("mlagents.trainers")
 
 
 class UnityPolicyException(UnityException):
diff --git a/ml-agents/mlagents/trainers/ppo/policy.py b/ml-agents/mlagents/trainers/ppo/policy.py
index 20504b6a63..d810e89eff 100644
--- a/ml-agents/mlagents/trainers/ppo/policy.py
+++ b/ml-agents/mlagents/trainers/ppo/policy.py
@@ -4,7 +4,7 @@
 from mlagents.trainers.ppo.models import PPOModel
 from mlagents.trainers.policy import Policy
 
-logger = logging.getLogger("unityagents")
+logger = logging.getLogger("mlagents.trainers")
 
 
 class PPOPolicy(Policy):