diff --git a/src/skywallet/skywallet_test.go b/src/skywallet/skywallet_test.go
index 7c6c295..5d9020f 100644
--- a/src/skywallet/skywallet_test.go
+++ b/src/skywallet/skywallet_test.go
@@ -346,7 +346,7 @@ func (suite *devicerSuit) TestTransactionSign() {
 	device := getMockDevice(driverMock)
 
 	// NOTE(denisacostaq@gmail.com): When
-	msg, err := device.TransactionSign(nil, nil)
+	msg, err := device.TransactionSign(nil, nil, true)
 
 	// NOTE(denisacostaq@gmail.com): Assert
 	suite.Nil(err)