diff --git a/tests/resources/reference_concept_from_image_schnell_heatmap.png b/tests/resources/reference_concept_from_image_schnell_heatmap.png
new file mode 100644
index 0000000..6ca1e4d
Binary files /dev/null and b/tests/resources/reference_concept_from_image_schnell_heatmap.png differ
diff --git a/tests/resources/reference_concept_schnell_heatmap.png b/tests/resources/reference_concept_schnell_heatmap.png
new file mode 100644
index 0000000..b491e0c
Binary files /dev/null and b/tests/resources/reference_concept_schnell_heatmap.png differ