diff --git a/official/projects/yolo/modeling/backbones/darknet_test.py b/official/projects/yolo/modeling/backbones/darknet_test.py
index 7f6e5798e146b4bb11db3a40999954bc24ed31ef..e775f78a76ff4119daa983acca65256e3855aebc 100644
--- a/official/projects/yolo/modeling/backbones/darknet_test.py
+++ b/official/projects/yolo/modeling/backbones/darknet_test.py
@@ -82,7 +82,10 @@ class DarknetTest(parameterized.TestCase, tf.test.TestCase):
 
     with strategy.scope():
       network = darknet.Darknet(
-          model_id='darknet53', min_size=3, max_size=5, use_sync_bn=use_sync_bn
+          model_id='darknet53',
+          min_level=3,
+          max_level=5,
+          use_sync_bn=use_sync_bn,
       )
       _ = network(inputs)