diff --git a/tools/bioimaging/bioimage_inference.xml b/tools/bioimaging/bioimage_inference.xml
index ee8618234e..fe27ba6398 100644
--- a/tools/bioimaging/bioimage_inference.xml
+++ b/tools/bioimaging/bioimage_inference.xml
@@ -56,6 +56,13 @@
             <output name="output_predicted_image" file="output_nucleisegboundarymodel.tif" compare="sim_size" delta="100" />
             <output name="output_predicted_image_matrix" file="output_nucleisegboundarymodel_matrix.npy" compare="sim_size" delta="100" />
         </test>
+        <test>
+            <param name="input_imaging_model" value="input_imaging_model.zip" location="https://zenodo.org/api/records/7768142/files/torchscript_tracing.pt/content"/>
+            <param name="input_image_file" value="input_bheu2024a.tif"/>
+            <param name="input_image_input_size" value="128,128,100,1"/>
+            <output name="output_predicted_image" file="output_bheu2024.tif" compare="sim_size" delta="100" />
+            <output name="output_predicted_image_matrix" file="output_bheu2024.npy" compare="sim_size" delta="100" />
+        </test>
     </tests>
     <help>
         <![CDATA[
diff --git a/tools/bioimaging/main.py b/tools/bioimaging/main.py
index 87a9c6ddee..b21fbda302 100644
--- a/tools/bioimaging/main.py
+++ b/tools/bioimaging/main.py
@@ -11,12 +11,7 @@
 
 def find_dim_order(user_in_shape, input_image):
     """
-    Find the correct order of input image's
-    shape. For a few models, the order of input size
-    mentioned in the RDF.yaml file is reversed compared
-    to the input image's original size. If it is reversed,
-    transpose the image to find correct order of image's
-    dimensions.
+    Find the correct order of input image's shape. For a few models, the order of input size mentioned in the RDF.yaml file is reversed compared to the input image's original size. If it is reversed, transpose the image to find correct order of image's dimensions.
     """
     image_shape = list(input_image.shape)
     # reverse the input shape provided from RDF.yaml file
diff --git a/tools/bioimaging/test-data/input_bheu2024a.tif b/tools/bioimaging/test-data/input_bheu2024a.tif
new file mode 100644
index 0000000000..fde9861483
Binary files /dev/null and b/tools/bioimaging/test-data/input_bheu2024a.tif differ