lib/chronoscope/nts/key_establishment_client.ex

@@ -33,12 +33,16 @@ defmodule Chronoscope.NTS.KeyEstablishmentClient do
   end
 
   defp perform_key_establishment(socket) do
-    :ok = @ssl.send(socket, KeyEstablishmentRequest.create())
+    with :ok <- @ssl.send(socket, KeyEstablishmentRequest.create()),
-    {:ok, peercert} = @ssl.peercert(socket)
+         {:ok, peercert} <- @ssl.peercert(socket) do
-
+      peercert
-    peercert
+      |> await_response()
-    |> await_response()
+      |> tap(fn _ -> @ssl.close(socket) end)
-    |> tap(fn _ -> @ssl.close(socket) end)
+    else
+      e ->
+        @ssl.close(socket)
+        {:error, e}
+    end
   end
 
   defp await_response(peercert) do