diff --git a/maths/gaussian_error_linear_unit.py b/maths/gaussian_error_linear_unit.py
index 7b5f875143b9..b3cbd7810716 100644
--- a/maths/gaussian_error_linear_unit.py
+++ b/maths/gaussian_error_linear_unit.py
@@ -13,7 +13,7 @@
 import numpy as np
 
 
-def sigmoid(vector: np.array) -> np.array:
+def sigmoid(vector: np.ndarray) -> np.ndarray:
     """
     Mathematical function sigmoid takes a vector x of K real numbers as input and
     returns 1/ (1 + e^-x).
@@ -25,17 +25,15 @@ def sigmoid(vector: np.array) -> np.array:
     return 1 / (1 + np.exp(-vector))
 
 
-def gaussian_error_linear_unit(vector: np.array) -> np.array:
+def gaussian_error_linear_unit(vector: np.ndarray) -> np.ndarray:
     """
     Implements the Gaussian Error Linear Unit (GELU) function
 
     Parameters:
-        vector (np.array): A  numpy array of shape (1,n)
-        consisting of real values
+        vector (np.ndarray): A  numpy array of shape (1, n) consisting of real values
 
     Returns:
-        gelu_vec (np.array): The input numpy array, after applying
-        gelu.
+        gelu_vec (np.ndarray): The input numpy array, after applying gelu
 
     Examples:
     >>> gaussian_error_linear_unit(np.array([-1.0, 1.0, 2.0]))