diff --git a/fastai2/callback/wandb.py b/fastai2/callback/wandb.py
index 4063e31e..c4d13828 100644
--- a/fastai2/callback/wandb.py
+++ b/fastai2/callback/wandb.py
@@ -45,7 +45,7 @@ def begin_fit(self):
 
             items = [self.dls.valid_ds.items[i] for i in idxs]
             test_tls = [tl._new(items, split_idx=1) for tl in self.dls.valid_ds.tls]
-            self.valid_dl = self.dls.valid_dl.new(Datasets(tls=test_tls), bs=self.n_preds)
+            self.valid_dl = DataLoader(self.dls.valid, bs=self.n_preds)
 
     def after_batch(self):
         "Log hyper-parameters and training loss"
diff --git a/nbs/70_callback.wandb.ipynb b/nbs/70_callback.wandb.ipynb
index 117741fd..201840ff 100644
--- a/nbs/70_callback.wandb.ipynb
+++ b/nbs/70_callback.wandb.ipynb
@@ -105,7 +105,7 @@
     "\n",
     "            items = [self.dls.valid_ds.items[i] for i in idxs]\n",
     "            test_tls = [tl._new(items, split_idx=1) for tl in self.dls.valid_ds.tls]\n",
-    "            self.valid_dl = self.dls.valid_dl.new(Datasets(tls=test_tls), bs=self.n_preds)\n",
+    "            self.valid_dl = DataLoader(self.dls.valid, bs=self.n_preds)\n",
     "\n",
     "    def after_batch(self):\n",
     "        \"Log hyper-parameters and training loss\"\n",