diff --git a/shard.yml b/shard.yml
index 6fbd105..6c6838d 100644
--- a/shard.yml
+++ b/shard.yml
@@ -1,5 +1,5 @@
 name: tensorflow_lite
-version: 1.2.0
+version: 1.2.1
 
 development_dependencies:
   ameba:
diff --git a/src/tensorflow_lite/client.cr b/src/tensorflow_lite/client.cr
index c9dbd5f..fffaf27 100644
--- a/src/tensorflow_lite/client.cr
+++ b/src/tensorflow_lite/client.cr
@@ -24,8 +24,8 @@ class TensorflowLite::Client
     @interpreter = Interpreter.new(@model, @options)
   end
 
-  def self.new(model : Bytes | Path | Model | String, threads : Int? = nil)
-    Client.new(model, threads) { |error_message| Log.warn { error_message } }
+  def self.new(model : Bytes | Path | Model | String, delegate : Delegate? = nil, threads : Int? = nil)
+    Client.new(model, delegate, threads) { |error_message| Log.warn { error_message } }
   end
 
   getter model : Model