diff --git a/02_production.ipynb b/02_production.ipynb
index 245f469..762c238 100644
--- a/02_production.ipynb
+++ b/02_production.ipynb
@@ -926,7 +926,8 @@
     }
    ],
    "source": [
-    "bears = bears.new(item_tfms=Resize(128), batch_tfms=aug_transforms(mult=2))\n",
+    "custom_aug_transforms = [RandomResizedCrop(128, min_scale=0.35), Flip(), Brightness(), Contrast(), Rotate(max_deg=10.0)]\n",
+    "bears = bears.new(item_tfms=Resize(128), batch_tfms=custom_aug_transforms)\n",
     "dls = bears.dataloaders(path)\n",
     "dls.train.show_batch(max_n=8, nrows=2, unique=True)"
    ]