diff --git a/python/trx/ctrl_if_bb.py b/python/trx/ctrl_if_bb.py
index 77e02d0..99c0451 100644
--- a/python/trx/ctrl_if_bb.py
+++ b/python/trx/ctrl_if_bb.py
@@ -72,12 +72,22 @@
 
 			return 0
 
+		# Gain control
 		elif self.verify_cmd(request, "SETRXGAIN", 1):
 			print("[i] Recv SETRXGAIN cmd")
 
 			# TODO: check gain value
 			gain = int(request[1])
-			self.tb.set_gain(gain)
+			self.tb.set_rx_gain(gain)
+
+			return 0
+
+		elif self.verify_cmd(request, "SETTXGAIN", 1):
+			print("[i] Recv SETTXGAIN cmd")
+
+			# TODO: check gain value
+			gain = int(request[1])
+			self.tb.set_tx_gain(gain)
 
 			return 0
 
diff --git a/python/trx/radio_if.py b/python/trx/radio_if.py
index 559c309..d8c0444 100644
--- a/python/trx/radio_if.py
+++ b/python/trx/radio_if.py
@@ -126,6 +126,10 @@
 		self.fc_set = True
 		self.fc = fc
 
-	def set_gain(self, gain):
+	def set_rx_gain(self, gain):
 		self.phy.set_gain(gain, 0)
-		self.gain = gain
+		self.rx_gain = gain
+
+	def set_tx_gain(self, gain):
+		# TODO: TX chain not implemented yet
+		self.tx_gain = gain