diff --git a/keras_segmentation/models/_pspnet_2.py b/keras_segmentation/models/_pspnet_2.py
index 2538b8b0d..4265ef5d4 100644
--- a/keras_segmentation/models/_pspnet_2.py
+++ b/keras_segmentation/models/_pspnet_2.py
@@ -34,11 +34,10 @@ def build(self, input_shape):
 
     def call(self, inputs, **kwargs):
         new_height, new_width = self.new_size
-        try:
-            resized = tf.image.resize(inputs, [new_height, new_width])
-        except AttributeError:
-            resized = tf.image.resize_images(inputs, [new_height, new_width],
-                                             align_corners=True)
+
+        resized = tf.compat.v1.image.resize(inputs, [new_height, new_width],
+            align_corners=True)
+        
         return resized
 
     def compute_output_shape(self, input_shape):