diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py
index 0e4b864c9c596ac34b47dea591b81ffcc36b5f29..49c0d7ad740e5fb61173d146784fb24def279933 100644
--- a/modules/deeplearning/srcnn_l1b_l2.py
+++ b/modules/deeplearning/srcnn_l1b_l2.py
@@ -514,10 +514,10 @@ class SRCNN:
     def test_step(self, mini_batch):
         inputs = [mini_batch[0]]
         labels = mini_batch[1]
-        in_nd = mini_batch.numpy()
+        in_nd = tf.make_ndarray(mini_batch[0])
         print('****: ', in_nd.shape, in_nd.min(), in_nd.max())
         pred = self.model(inputs, training=False)
-        in_nd = pred.numpy()
+        in_nd = tf.make_ndarray(pred)
         print('****: ', in_nd.shape, in_nd.min(), in_nd.max())
         t_loss = self.loss(labels, pred)