diff --git a/priv/grpcbox_service_bhvr.erl b/priv/grpcbox_service_bhvr.erl
index 6aef135..92629b2 100644
--- a/priv/grpcbox_service_bhvr.erl
+++ b/priv/grpcbox_service_bhvr.erl
@@ -8,10 +8,10 @@
 -module({{module_name}}_bhvr).
 
 {{#methods}}
--callback {{method}}({{^input_stream}}{{#output_stream}}{{pb_module}}:{{input}}(), grpcbox_stream:t(){{/output_stream}}{{/input_stream}}{{#input_stream}}{{^output_stream}}reference(), grpcbox_stream:t(){{/output_stream}}{{#output_stream}}reference(), grpcbox_stream:t(){{/output_stream}}{{/input_stream}}{{^input_stream}}{{^output_stream}}ctx:ctx(), {{pb_module}}:{{input}}(){{/output_stream}}{{/input_stream}}) ->
+-callback {{method}}({{^input_stream}}{{#output_stream}}{{pb_module}}:{{input}}(), grpcbox_stream:t(){{/output_stream}}{{/input_stream}}{{#input_stream}}{{^output_stream}}{{pb_module}}:{{input}}(), grpcbox_stream:t(){{/output_stream}}{{#output_stream}}{{pb_module}}:{{input}}(), grpcbox_stream:t(){{/output_stream}}{{/input_stream}}{{^input_stream}}{{^output_stream}}ctx:ctx(), {{pb_module}}:{{input}}(){{/output_stream}}{{/input_stream}}) ->
     {{#output_stream}}ok{{/output_stream}}{{^output_stream}}{ok, {{pb_module}}:{{output}}(), ctx:ctx()}{{/output_stream}}{{#output_stream}} | {ok, grpcbox_stream:t()}{{/output_stream}}{{#output_stream}} | {ok, {{pb_module}}:{{output}}(), grpcbox_stream:t()}{{/output_stream}}{{#output_stream}} | {stop, grpcbox_stream:t()}{{/output_stream}}{{#output_stream}} | {stop, {{pb_module}}:{{output}}(), grpcbox_stream:t()}{{/output_stream}} | grpcbox_stream:grpc_error_response().
 {{/methods}}
 
 -callback init(atom(), grpcbox_stream:t()) -> grpcbox_stream:t().
 -callback handle_info(any(), grpcbox_stream:t()) -> grpcbox_stream:t().
--optional_callbacks([init/2, handle_info/2]).
\ No newline at end of file
+-optional_callbacks([init/2, handle_info/2]).