diff --git a/metastable_baselines/ppo/policies.py b/metastable_baselines/ppo/policies.py
index 8bcb2c0..108108b 100644
--- a/metastable_baselines/ppo/policies.py
+++ b/metastable_baselines/ppo/policies.py
@@ -25,7 +25,7 @@ from stable_baselines3.common.torch_layers import (
     MlpExtractor,
     NatureCNN,
 )
-from stable_baselines3.common.type_aliases import Schedules
+from stable_baselines3.common.type_aliases import Schedule
 
 from stable_baselines3.common.policies import BasePolicy
 from stable_baselines3.common.torch_layers import (