diff --git a/ppgan/models/generators/makeup.py b/ppgan/models/generators/makeup.py
index 0532e32ae01b38b9b428219e22bc0e81a993cf7b..5f80ae4f4106f6f0bb07af840757d2b02ca3685d 100644
--- a/ppgan/models/generators/makeup.py
+++ b/ppgan/models/generators/makeup.py
@@ -207,7 +207,7 @@ class TNetDown(paddle.nn.Layer):
         return out
 
 
-class GetMatrix(paddle.fluid.dygraph.Layer):
+class GetMatrix(paddle.nn.Layer):
     def __init__(self, dim_in, dim_out):
         super(GetMatrix, self).__init__()
         self.get_gamma = nn.Conv2D(dim_in,