diff --git a/src/inception_model.py b/src/inception_model.py
index 09e032ca70a8e63d9459464e968d25fac29cd4f1..64b5e09674dd9fccd8282b80c4fb9e87bbd5ea5c 100644
--- a/src/inception_model.py
+++ b/src/inception_model.py
@@ -124,7 +124,7 @@ class InceptionModelBase:
         tower = pooling(pool_kernel, strides=(1, 1), padding='same', name=block_name+block_type)(input_x)
 
         # convolution block
-        tower = Conv2D(tower_filter, (1, 1), padding='same', name=block_name+"1x1")(tower)
+        tower = layers.Conv2D(tower_filter, (1, 1), padding='same', name=block_name+"1x1")(tower)
         tower = self.act(tower, activation, **act_settings)
 
         return tower