diff --git a/DESCRIPTION b/DESCRIPTION
index 2511d6b..081925d 100644
--- a/DESCRIPTION
+++ b/DESCRIPTION
@@ -1,6 +1,6 @@
 Package: diffuseR
 Title: Functional Interface to Diffusion Models in R
-Version: 0.0.1
+Version: 0.0.2
 Authors@R: 
     person("Troy", "Hernandez", email = "troy@cornball.ai", role = c("aut", "cre"))
 Description: A native R implementation of diffusion models providing a functional 
diff --git a/R/img2img.R b/R/img2img.R
index 989febb..6fbcdd5 100644
--- a/R/img2img.R
+++ b/R/img2img.R
@@ -60,6 +60,7 @@ img2img <- function (input_image, prompt, negative_prompt = NULL,
 
     if (is.null(pipeline)) {
         pipeline <- load_pipeline(model_name = model_name, m2d = m2d,
+            i2i = TRUE,
             unet_dtype_str = unet_dtype_str,
             use_native_decoder = use_native_decoder,
             use_native_text_encoder = use_native_text_encoder,