diff --git a/ppdet/optimizer.py b/ppdet/optimizer.py
index 87c605f6813c74dac7a242f4df7d92dd5db690e7..82be7535a90e46d73927870a30492e7e6518dab3 100644
--- a/ppdet/optimizer.py
+++ b/ppdet/optimizer.py
@@ -61,7 +61,7 @@ class PiecewiseDecay(object):
             for i in self.gamma:
                 value.append(base_lr * i)
 
-        return optimizer.lr_scheduler.PiecewiseLR(boundary, value)
+        return optimizer.lr.PiecewiseDecay(boundary, value)
 
 
 @serializable