From e8365f5f434237389849a1a7b0cd4b24d9c997a7 Mon Sep 17 00:00:00 2001
From: akhilkumarpilli <akhilkumar7947@gmail.com>
Date: Tue, 27 Aug 2024 14:53:55 +0530
Subject: [PATCH] address comments

---
 tests/systemtests/staking_test.go | 1 -
 x/auth/ante/fee.go                | 1 +
 2 files changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/systemtests/staking_test.go b/tests/systemtests/staking_test.go
index 0b18408410df..f197bff1a0ac 100644
--- a/tests/systemtests/staking_test.go
+++ b/tests/systemtests/staking_test.go
@@ -10,7 +10,6 @@ import (
 )
 
 func TestStakeUnstake(t *testing.T) {
-	t.Skip("The fee deduction is not yet implemented in v2")
 	// Scenario:
 	// delegate tokens to validator
 	// undelegate some tokens
diff --git a/x/auth/ante/fee.go b/x/auth/ante/fee.go
index 3082fec2baa3..e2a318c2dc61 100644
--- a/x/auth/ante/fee.go
+++ b/x/auth/ante/fee.go
@@ -76,6 +76,7 @@ func (dfd DeductFeeDecorator) AnteHandle(ctx sdk.Context, tx sdk.Tx, _ bool, nex
 }
 
 // ValidateTx implements an TxValidator for DeductFeeDecorator
+// Note: This method is applicable only for transactions that implement the sdk.FeeTx interface.
 func (dfd DeductFeeDecorator) ValidateTx(ctx context.Context, tx sdk.Tx) error {
 	feeTx, ok := tx.(sdk.FeeTx)
 	if !ok {