trx/radio_if.py: fix RX / TX gain control
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