diff --git a/tests/helpers/image_generation_test_helper.py b/tests/helpers/image_generation_test_helper.py
index 7bcdc8a..d1d8929 100644
--- a/tests/helpers/image_generation_test_helper.py
+++ b/tests/helpers/image_generation_test_helper.py
@@ -63,7 +63,7 @@ class ImageGeneratorTestHelper:
                 os.remove(output_image_path)
 
     @staticmethod
-    def resolve_path(path) -> Path:
+    def resolve_path(path) -> Path | None:
         if path is None:
             return None
         return Path(__file__).parent.parent / "resources" / path
diff --git a/tests/resources/reference_dev_image_to_image_init.png b/tests/resources/reference_dev_image_to_image_init.png
deleted file mode 100644
index 21c7781..0000000
Binary files a/tests/resources/reference_dev_image_to_image_init.png and /dev/null differ
diff --git a/tests/resources/reference_dev_image_to_image_result.png b/tests/resources/reference_dev_image_to_image_result.png
index e52531b..6fd8baa 100644
Binary files a/tests/resources/reference_dev_image_to_image_result.png and b/tests/resources/reference_dev_image_to_image_result.png differ
diff --git a/tests/test_generate_image.py b/tests/test_generate_image.py
index 34e5f8d..d234361 100644
--- a/tests/test_generate_image.py
+++ b/tests/test_generate_image.py
@@ -46,13 +46,13 @@ class TestImageGenerator:
     def test_image_generation_dev_image_to_image(self):
         ImageGeneratorTestHelper.assert_matches_reference_image(
             reference_image_path="reference_dev_image_to_image_result.png",
-            init_image_path="reference_dev_image_to_image_init.png",
-            init_image_strength=0.126,  # fill 1/8 steps
+            init_image_path="reference_dev_lora.png",
+            init_image_strength=0.4,
             output_image_path=TestImageGenerator.OUTPUT_IMAGE_FILENAME,
             model_config=ModelConfig.FLUX1_DEV,
             steps=8,
-            seed=42,
-            height=768,
+            seed=44,
+            height=341,
             width=768,
-            prompt="astronauts in a jungle",
+            prompt="Luxury food photograph of a burger",
         )