diff --git a/test.py b/test.py
index b59de153a3ff0891546d0233315be315e215d19c..a176f1b9f26a12f4041550cacedd7d5b08e6eff5 100644
--- a/test.py
+++ b/test.py
@@ -1,8 +1,8 @@
-from nets.yolo3 import yolo_body
+from nets.yolo4 import yolo_body
 from keras.layers import Input
 inputs = Input([416,416,3])
 model = yolo_body(inputs,3,80)
 model.summary()
 
 for i,layer in enumerate(model.layers):
-    print(i,layer.name)
\ No newline at end of file
+    print(i,layer.name)