diff --git a/deepspeech/__init__.py b/deepspeech/__init__.py
index fbec5a5e8c8e5537f8689b3e7da2b1e6eb34f1a3..d85a3dde7d44a388878a0b0f411f4a2bd594800d 100644
--- a/deepspeech/__init__.py
+++ b/deepspeech/__init__.py
@@ -362,7 +362,7 @@ class GLU(nn.Layer):
         self.dim = dim
 
     def forward(self, xs):
-        return F.glu(xs, dim=self.dim)
+        return F.glu(xs, axis=self.dim)
 
 
 if not hasattr(paddle.nn, 'GLU'):