diff --git a/efficientnet/model.py b/efficientnet/model.py
index 0913c54..18b55d2 100644
--- a/efficientnet/model.py
+++ b/efficientnet/model.py
@@ -64,7 +64,7 @@ def __call__(self, shape, dtype=K.floatx(), partition_info=None):
         del partition_info
         kernel_height, kernel_width, _, out_filters = shape
         fan_out = int(kernel_height * kernel_width * out_filters)
-        return tf.random_normal(
+        return tf.random.normal(
             shape, mean=0.0, stddev=np.sqrt(2.0 / fan_out), dtype=dtype)
 
 class DenseKernalInitializer(Initializer):