diff --git a/docs/load-rumi-jawi.ipynb b/docs/load-rumi-jawi.ipynb
index 08867d59..20f667ad 100644
--- a/docs/load-rumi-jawi.ipynb
+++ b/docs/load-rumi-jawi.ipynb
@@ -67,8 +67,8 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "CPU times: user 5.66 s, sys: 1.09 s, total: 6.75 s\n",
-      "Wall time: 8.54 s\n"
+      "CPU times: user 5.81 s, sys: 1.24 s, total: 7.06 s\n",
+      "Wall time: 9.8 s\n"
      ]
     }
    ],
@@ -95,7 +95,7 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "INFO:malaya.jawi_rumi:tested on first 10k Jawi-Rumi test set, dataset at https://huggingface.co/datasets/mesolitica/rumi-jawi\n"
+      "INFO:malaya.rumi_jawi:tested on first 10k Rumi-Jawi test set, dataset at https://huggingface.co/datasets/mesolitica/rumi-jawi\n"
      ]
     },
     {
@@ -131,16 +131,16 @@
        "      <th>small</th>\n",
        "      <td>42.7</td>\n",
        "      <td>13.1</td>\n",
-       "      <td>0.004477</td>\n",
-       "      <td>0.013642</td>\n",
+       "      <td>0.000617</td>\n",
+       "      <td>0.001928</td>\n",
        "      <td>256.0</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>base</th>\n",
        "      <td>234.0</td>\n",
        "      <td>63.8</td>\n",
-       "      <td>0.000764</td>\n",
-       "      <td>0.003042</td>\n",
+       "      <td>0.000124</td>\n",
+       "      <td>0.000438</td>\n",
        "      <td>256.0</td>\n",
        "    </tr>\n",
        "  </tbody>\n",
@@ -149,8 +149,8 @@
       ],
       "text/plain": [
        "       Size (MB)  Quantized Size (MB)       CER       WER  Suggested length\n",
-       "small       42.7                 13.1  0.004477  0.013642             256.0\n",
-       "base       234.0                 63.8  0.000764  0.003042             256.0"
+       "small       42.7                 13.1  0.000617  0.001928             256.0\n",
+       "base       234.0                 63.8  0.000124  0.000438             256.0"
       ]
      },
      "execution_count": 3,
@@ -159,7 +159,7 @@
     }
    ],
    "source": [
-    "malaya.jawi_rumi.available_transformer()"
+    "malaya.rumi_jawi.available_transformer()"
    ]
   },
   {
diff --git a/example/rumi-jawi/load-rumi-jawi.ipynb b/example/rumi-jawi/load-rumi-jawi.ipynb
index 08867d59..20f667ad 100644
--- a/example/rumi-jawi/load-rumi-jawi.ipynb
+++ b/example/rumi-jawi/load-rumi-jawi.ipynb
@@ -67,8 +67,8 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "CPU times: user 5.66 s, sys: 1.09 s, total: 6.75 s\n",
-      "Wall time: 8.54 s\n"
+      "CPU times: user 5.81 s, sys: 1.24 s, total: 7.06 s\n",
+      "Wall time: 9.8 s\n"
      ]
     }
    ],
@@ -95,7 +95,7 @@
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "INFO:malaya.jawi_rumi:tested on first 10k Jawi-Rumi test set, dataset at https://huggingface.co/datasets/mesolitica/rumi-jawi\n"
+      "INFO:malaya.rumi_jawi:tested on first 10k Rumi-Jawi test set, dataset at https://huggingface.co/datasets/mesolitica/rumi-jawi\n"
      ]
     },
     {
@@ -131,16 +131,16 @@
        "      <th>small</th>\n",
        "      <td>42.7</td>\n",
        "      <td>13.1</td>\n",
-       "      <td>0.004477</td>\n",
-       "      <td>0.013642</td>\n",
+       "      <td>0.000617</td>\n",
+       "      <td>0.001928</td>\n",
        "      <td>256.0</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>base</th>\n",
        "      <td>234.0</td>\n",
        "      <td>63.8</td>\n",
-       "      <td>0.000764</td>\n",
-       "      <td>0.003042</td>\n",
+       "      <td>0.000124</td>\n",
+       "      <td>0.000438</td>\n",
        "      <td>256.0</td>\n",
        "    </tr>\n",
        "  </tbody>\n",
@@ -149,8 +149,8 @@
       ],
       "text/plain": [
        "       Size (MB)  Quantized Size (MB)       CER       WER  Suggested length\n",
-       "small       42.7                 13.1  0.004477  0.013642             256.0\n",
-       "base       234.0                 63.8  0.000764  0.003042             256.0"
+       "small       42.7                 13.1  0.000617  0.001928             256.0\n",
+       "base       234.0                 63.8  0.000124  0.000438             256.0"
       ]
      },
      "execution_count": 3,
@@ -159,7 +159,7 @@
     }
    ],
    "source": [
-    "malaya.jawi_rumi.available_transformer()"
+    "malaya.rumi_jawi.available_transformer()"
    ]
   },
   {