diff --git a/tensorflow_addons/layers/normalizations.py b/tensorflow_addons/layers/normalizations.py
index 3975aa9777..7322701f5f 100644
--- a/tensorflow_addons/layers/normalizations.py
+++ b/tensorflow_addons/layers/normalizations.py
@@ -75,7 +75,7 @@ def __init__(
         self,
         groups: int = 2,
         axis: int = -1,
-        epsilon: int = 1e-3,
+        epsilon: float = 1e-3,
         center: bool = True,
         scale: bool = True,
         beta_initializer: types.Initializer = "zeros",