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