diff --git a/examples/cli/main.cpp b/examples/cli/main.cpp
index 0586fcd6..7dce1c96 100644
--- a/examples/cli/main.cpp
+++ b/examples/cli/main.cpp
@@ -960,7 +960,7 @@ int main(int argc, const char* argv[]) {
                           params.style_ratio,
                           params.normalize_input,
                           params.input_id_images_path.c_str(),
-                          step_callback);
+                          (step_callback_t)step_callback);
     } else {
         sd_image_t input_image = {(uint32_t)params.width,
                                   (uint32_t)params.height,
diff --git a/stable-diffusion.h b/stable-diffusion.h
index cb13a028..2da0dec3 100644
--- a/stable-diffusion.h
+++ b/stable-diffusion.h
@@ -146,7 +146,7 @@ SD_API sd_ctx_t* new_sd_ctx(const char* model_path,
 
 SD_API void free_sd_ctx(sd_ctx_t* sd_ctx);
 
-typedef void (*step_callback_t)(int, struct ggml_tensor*, enum SDVersion);
+typedef void (*step_callback_t)(int, struct ggml_tensor*, int);
 
 SD_API sd_image_t* txt2img(sd_ctx_t* sd_ctx,
                            const char* prompt,