diff --git a/vgg16.py b/vgg16.py
index 373c5c2..b2a959a 100644
--- a/vgg16.py
+++ b/vgg16.py
@@ -99,7 +99,7 @@ def VGG16(include_top=True, weights='imagenet',
                                       default_size=224,
                                       min_size=48,
                                       data_format=K.image_data_format(),
-                                      include_top=include_top)
+                                      require_flatten=include_top)
 
     if input_tensor is None:
         img_input = Input(shape=input_shape)