diff --git a/skluc/data/mldatasets/ImageDataset.py b/skluc/data/mldatasets/ImageDataset.py
index b8f6ee6d0a3f10cee9ba659785ec4b4514d5809d..904f4fd226faa381bbcfdaead6d14bf8480376de 100644
--- a/skluc/data/mldatasets/ImageDataset.py
+++ b/skluc/data/mldatasets/ImageDataset.py
@@ -79,6 +79,7 @@ class ImageDataset(Dataset):
                 length_by_chanel = int(length_by_chanel)
             images_mat = np.reshape(images_vec, (images_vec.shape[0], length_by_chanel, self.DEPTH), order='F')
             images = np.reshape(images_mat, (images_mat.shape[0], self.HEIGHT, self.WIDTH, self.DEPTH), order='C')
+            logger.debug("Images mat shape: {}".format(images.shape))
             setattr(self, kw, LabeledData(images, labels))
 
     def flatten(self):
diff --git a/skluc/test/test_transformation/TestResizeTransformer.py b/skluc/test/test_transformation/TestResizeTransformer.py
index 60ea6c000217edd2c4f26deec912feecdf4c95ad..23ba8717fa068cbc6c1d02c4b7d063439cf0ad65 100644
--- a/skluc/test/test_transformation/TestResizeTransformer.py
+++ b/skluc/test/test_transformation/TestResizeTransformer.py
@@ -33,6 +33,9 @@ class TestResizeTransformer(unittest.TestCase):
                 d.to_image()
                 trans = ResizeTransformer(data_name=data_name, output_shape=size)
                 d.apply_transformer(transformer=trans)
+                new_size = d.train.data[0].shape[:-1]
+                self.assertEqual(new_size[0], size[0])
+                self.assertEqual(new_size[1], size[1])
                 del trans
 
     def test_init(self):