throw correct error when pair address is invalid

Cardinal-Cryptography · Sep 25, 2024 · 58583b8 · 58583b8
1 parent f2e9ea8
commit 58583b8
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/amm/contracts/router_v2/lib.rs b/amm/contracts/router_v2/lib.rs
@@ -76,9 +76,9 @@ pub mod router_v2 {
                 Some(pool_id) => pool_id,
                 None => self.pair_factory_ref().create_pair(token_0, token_1)?,
             };
-            match self.get_pool(pool_id) {
-                Ok(Pool::Pair(pair)) => Ok(pair),
-                _ => Err(RouterV2Error::PairNotFound),
+            match self.get_pool(pool_id)? {
+                Pool::Pair(pair) => Ok(pair),
+                Pool::StablePool(_) => Err(RouterV2Error::PairNotFound),
             }
         }