diff --git a/dygraph/models/unet.py b/dygraph/models/unet.py
index b55e3614b6988a0102eb3e6f17093e59673eae70..6e04c8b2f17aeca763dc9653b6b2da73835979c7 100644
--- a/dygraph/models/unet.py
+++ b/dygraph/models/unet.py
@@ -13,7 +13,8 @@
 # limitations under the License.
 
 import paddle.fluid as fluid
-from paddle.fluid.dygraph import Conv2D, BatchNorm, Pool2D
+from paddle.fluid.dygraph import Conv2D, Pool2D
+from paddle.fluid.dygraph import SyncBatchNorm as BatchNorm
 
 
 class UNet(fluid.dygraph.Layer):