diff --git a/ML/TensorFlow/Basics/tutorial17-tensorboard/2_tb_scalars.py b/ML/TensorFlow/Basics/tutorial17-tensorboard/2_tb_scalars.py
index 436378ed..975c1e61 100644
--- a/ML/TensorFlow/Basics/tutorial17-tensorboard/2_tb_scalars.py
+++ b/ML/TensorFlow/Basics/tutorial17-tensorboard/2_tb_scalars.py
@@ -52,9 +52,9 @@ def augment(image, label):
 ds_train = ds_train.prefetch(AUTOTUNE)
 
 # Setup for test Dataset
-ds_test = ds_train.map(normalize_img, num_parallel_calls=AUTOTUNE)
-ds_test = ds_train.batch(BATCH_SIZE)
-ds_test = ds_train.prefetch(AUTOTUNE)
+ds_test = ds_test.map(normalize_img, num_parallel_calls=AUTOTUNE)
+ds_test = ds_test.batch(BATCH_SIZE)
+ds_test = ds_test.prefetch(AUTOTUNE)
 
 class_names = [
     "Airplane",