diff --git a/src/beignet/special/_dawson_integral_f.py b/src/beignet/special/_dawson_integral_f.py
index 07db8de2c6..a6aed878cc 100644
--- a/src/beignet/special/_dawson_integral_f.py
+++ b/src/beignet/special/_dawson_integral_f.py
@@ -6,5 +6,5 @@
 from ._error_erfi import error_erfi
 
 
-def dawson_integral_f(z: Tensor) -> Tensor:
-    return math.sqrt(torch.pi) / 2.0 * torch.exp(-torch.square(z)) * error_erfi(z)
+def dawson_integral_f(input: Tensor) -> Tensor:
+    return math.sqrt(torch.pi) / 2.0 * torch.exp(-(input**2)) * error_erfi(input)