diff --git a/models/olive_oil/olive_oil-transformer.ipynb b/models/olive_oil/olive_oil-transformer.ipynb
index 734af65..adb8a6b 100644
--- a/models/olive_oil/olive_oil-transformer.ipynb
+++ b/models/olive_oil/olive_oil-transformer.ipynb
@@ -1421,6 +1421,7 @@
     "        decay_steps=7000    # Aumentato per permettere più esplorazione\n",
     "    )\n",
     "\n",
+    "    @keras.saving.register_keras_serializable()\n",
     "    def weighted_huber_loss(y_true, y_pred):\n",
     "        # Pesi per diversi output\n",
     "        weights = tf.constant([1.0, 0.8, 0.8, 1.0, 0.6], dtype=tf.float32)\n",