diff --git a/demo/image_matting/colab/inference.py b/demo/image_matting/colab/inference.py
index 2cdaff7..c4f280b 100644
--- a/demo/image_matting/colab/inference.py
+++ b/demo/image_matting/colab/inference.py
@@ -90,7 +90,7 @@ if __name__ == '__main__':
         im = F.interpolate(im, size=(im_rh, im_rw), mode='area')
 
         # inference
-        _, _, matte = modnet(im.cuda(), inference=False)
+        _, _, matte = modnet(im.cuda(), True)
 
         # resize and save matte
         matte = F.interpolate(matte, size=(im_h, im_w), mode='area')
diff --git a/demo/video_matting/webcam/run.py b/demo/video_matting/webcam/run.py
index 79a482d..4b7a5f9 100644
--- a/demo/video_matting/webcam/run.py
+++ b/demo/video_matting/webcam/run.py
@@ -52,7 +52,7 @@ while(True):
         frame_tensor = frame_tensor.cuda()
     
     with torch.no_grad():
-        _, _, matte_tensor = modnet(frame_tensor, inference=True)
+        _, _, matte_tensor = modnet(frame_tensor, True)
 
     matte_tensor = matte_tensor.repeat(1, 3, 1, 1)
     matte_np = matte_tensor[0].data.cpu().numpy().transpose(1, 2, 0)