--- a/qupath_roi_splitter.xml	Fri Jul 19 14:33:33 2024 +0000
+++ b/qupath_roi_splitter.xml	Tue Jul 30 12:59:05 2024 +0000
@@ -1,7 +1,7 @@
 <tool id="qupath_roi_splitter" name="QuPath ROI Splitter" version="@VERSION@+galaxy@VERSION_SUFFIX@">
     <description>Split ROI coordinates of QuPath TMA annotation by cell type (classification)</description>
     <macros>
-        <token name="@VERSION@">0.3.0</token>
+        <token name="@VERSION@">0.3.1</token>
         <token name="@VERSION_SUFFIX@">0</token>
     </macros>
     <requirements>
@@ -21,7 +21,9 @@
         #end for
         mkdir out
         && mv *.txt out/
+        #if $optional.img
         && mv *.png out/
+        #end if
     ]]></command>
     <inputs>
         <param name="input_collection" type="data_collection" format="geojson" label="Input QuPath annotation" help="Collection containing GeoJSON files"/>
@@ -33,14 +35,15 @@
     </inputs>
     <outputs>
         <collection name="output_txts" type="list" label="${tool.name} on ${on_string}: ROI data">
-            <discover_datasets pattern="(?P&lt;name&gt;.*\.txt)" directory="out" visible="false" ext="txt"/>
+            <discover_datasets pattern="(?P&lt;name&gt;.*\.txt)" directory="out" visible="false" ext="tabular"/>
         </collection>
         <collection name="output_imgs" type="list" label="${tool.name} on ${on_string}: Images of ROIs">
+            <filter>optional['img']</filter>
             <discover_datasets pattern="(?P&lt;name&gt;.*\.png)" directory="out" visible="false" ext="png"/>
         </collection>
     </outputs>
     <tests>
-        <test>
+        <test expect_num_outputs="2">
             <param name="input_collection">
                 <collection type="list">
                     <element name="annotations_TMA_E-5.geojson" value="annotations_TMA_E-5.geojson" />