diff --git a/app/custom_proposal_handler_test.go b/app/custom_proposal_handler_test.go
index 86b50bcd19..4cbc25b768 100644
--- a/app/custom_proposal_handler_test.go
+++ b/app/custom_proposal_handler_test.go
@@ -1,3 +1,5 @@
+// Forked from here and extended to test ethermint txs https://github.com/cosmos/cosmos-sdk/blob/v0.47.10/baseapp/abci_utils_test.go
+// TODO: remove this once cosmos is upgraded: https://github.com/zeta-chain/node/issues/2156
 package app_test
 
 import (
diff --git a/app/mempool/priority_nonce_mempool.go b/app/mempool/priority_nonce_mempool.go
index cb6ed8748f..e6d61ca91f 100644
--- a/app/mempool/priority_nonce_mempool.go
+++ b/app/mempool/priority_nonce_mempool.go
@@ -1,4 +1,4 @@
-// This proposal handler is taken from https://github.com/cosmos/cosmos-sdk/blob/v0.47.10/types/mempool/priority_nonce.go
+// This mempool implementation is taken from https://github.com/cosmos/cosmos-sdk/blob/v0.47.10/types/mempool/priority_nonce.go
 // Only difference is extraction of senders and nonce from tx. In latest version of cosmos, there is a way to provide adapter for this, but in 0.47.10 this is the only way.
 // TODO: remove this once cosmos is upgraded: https://github.com/zeta-chain/node/issues/2156
 
diff --git a/app/mempool/priority_nonce_mempool_test.go b/app/mempool/priority_nonce_mempool_test.go
index 4ad8cbf5bb..3fa1848df5 100644
--- a/app/mempool/priority_nonce_mempool_test.go
+++ b/app/mempool/priority_nonce_mempool_test.go
@@ -1,3 +1,5 @@
+// Forked from here and extended to test ethermint txs https://github.com/cosmos/cosmos-sdk/blob/v0.47.10/types/mempool/priority_nonce_test.go
+// TODO: remove this once cosmos is upgraded: https://github.com/zeta-chain/node/issues/2156
 package mempool_test
 
 import (