diff --git a/src/amoc_rest_api.erl b/src/amoc_rest_api.erl
index 1d07021..eb19f96 100644
--- a/src/amoc_rest_api.erl
+++ b/src/amoc_rest_api.erl
@@ -103,7 +103,7 @@ prepare_validator(SchemaVer) ->
     prepare_validator(get_openapi_path(), SchemaVer).
 
 -doc """
-Loads the JSON schema and the desired validation draft into a `t:jesse_state:state()`.
+Loads the JSON schema and the desired validation draft into a `t:jesse_state:state/0`.
 """.
 -spec prepare_validator(file:name_all(), binary()) -> jesse_state:state().
 prepare_validator(OpenApiPath, SchemaVer) ->
diff --git a/src/amoc_rest_logic_handler.erl b/src/amoc_rest_logic_handler.erl
index 4db0e2b..8937f18 100644
--- a/src/amoc_rest_logic_handler.erl
+++ b/src/amoc_rest_logic_handler.erl
@@ -34,7 +34,7 @@
     {accept_callback_return(), cowboy_req:req(), context()}.
 
 -callback provide_callback(amoc_rest_api:class(), amoc_rest_api:operation_id(), cowboy_req:req(), context()) ->
-    {cowboy_req:resp_body(), cowboy_req:req(), context()}.
+    {provide_callback_return(), cowboy_req:req(), context()}.
 
 -export([api_key_callback/2, accept_callback/4, provide_callback/4]).
 -ignore_xref([api_key_callback/2, accept_callback/4, provide_callback/4]).