1 files changed, 2 insertions, 2 deletions
diff --git a/main/Main.hs b/main/Main.hs
index e055275..59ea7c3 100644
--- a/main/Main.hs
+++ b/main/Main.hs
@@ -893,14 +893,14 @@ cmdDiscoveryInit = void $ do
 
 cmdDiscovery :: Command
 cmdDiscovery = void $ do
-    Just peer <- gets csIcePeer
+    server <- asks ciServer
     st <- getStorage
     sref <- asks ciLine
     eprint <- asks ciPrint
     liftIO $ readRef st (BC.pack sref) >>= \case
         Nothing -> error "ref does not exist"
         Just ref -> do
-            res <- runExceptT $ sendToPeer peer $ DiscoverySearch ref
+            res <- runExceptT $ discoverySearch server ref
             case res of
                  Right _ -> return ()
                  Left err -> eprint err