diff --git a/cifar_test_sub_image.ipynb b/cifar_test_sub_image.ipynb
index c431df5..df21a55 100644
--- a/cifar_test_sub_image.ipynb
+++ b/cifar_test_sub_image.ipynb
@@ -33,7 +33,7 @@
     "from rbm.entity import Entity, EntityParams, TrainingParams\n",
     "from rbm.matrix import Mat, np, rms_error_accu, sample_gaussian\n",
     "from rbm.torch import Optimizer\n",
-    "from image.sub_image import SubImageExtract, SubImageExtractCombine, normalize\n",
+    "from image.sub_image import SubImageExtract, SubImageCombine, normalize\n",
     "import math\n",
     "import random"
    ],
@@ -387,7 +387,7 @@
     }
    },
    "source": [
-    "combiner = SubImageExtractCombine(PATCH, PATCH, STRIDE, STRIDE, 32, 32, n_ch=3)\n",
+    "combiner = SubImageCombine(PATCH, PATCH, STRIDE, STRIDE, 32, 32, n_ch=3)\n",
     "n_images = n_side\n",
     "nx_steps = (32 - PATCH) // STRIDE + 1\n",
     "ny_steps = (32 - PATCH) // STRIDE + 1\n",