diff --git a/deepspeech/__init__.py b/deepspeech/__init__.py
index fbec5a5e8..d85a3dde7 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'):