diff --git a/dataset/dataset.py b/dataset/dataset.py
index d098ead..6b4489b 100644
--- a/dataset/dataset.py
+++ b/dataset/dataset.py
@@ -54,7 +54,7 @@ class Dataset:
 
     def _get_image(self, image_path):
         img = tf.io.read_file(image_path)
-        img = tf.io.decode_jpeg(img, channels=3) / 255
+        img = tf.io.decode_jpeg(img, channels=3) / 255.
         return tf.cast(img, dtype=tf.float32, name=None)
 
     def _preprocess(self, image_path):