diff --git a/src/enoise_connection.erl b/src/enoise_connection.erl
index 215a937..53c12dd 100644
--- a/src/enoise_connection.erl
+++ b/src/enoise_connection.erl
@@ -57,7 +57,7 @@ start_link(TcpSock, Rx, Tx, Owner, {Active0, Buf}) ->
 send(Noise, Data) ->
     gen_server:call(Noise, {send, Data}).
 
--spec set_active(Noise :: pid(), Active :: boolean()) -> ok | {error, term()}.
+-spec set_active(Noise :: pid(), Active :: true | once) -> ok | {error, term()}.
 set_active(Noise, Active) ->
     gen_server:call(Noise, {active, self(), Active}).
 
diff --git a/src/enoise_hs_state.erl b/src/enoise_hs_state.erl
index dff6f2d..301a936 100644
--- a/src/enoise_hs_state.erl
+++ b/src/enoise_hs_state.erl
@@ -90,6 +90,7 @@ read_message(HS = #noise_hs{ msgs = [{in, Msg} | Msgs] }, Message) ->
         Err = {error, _} -> Err
     end.
 
+-spec remote_keys(HS :: state()) -> keypair().
 remote_keys(#noise_hs{ rs = RS }) ->
     RS.