diff --git a/ml-agents/mlagents/trainers/tests/test_rl_trainer.py b/ml-agents/mlagents/trainers/tests/test_rl_trainer.py
index b912d8ff47..8d5c1297d4 100644
--- a/ml-agents/mlagents/trainers/tests/test_rl_trainer.py
+++ b/ml-agents/mlagents/trainers/tests/test_rl_trainer.py
@@ -1,4 +1,4 @@
-import unittest.mock as mock
+from unittest import mock
 import yaml
 import mlagents.trainers.tests.mock_brain as mb
 import numpy as np