diff --git a/fink_science/random_forest_snia/processor.py b/fink_science/random_forest_snia/processor.py
index 131cc2b6..d553997e 100644
--- a/fink_science/random_forest_snia/processor.py
+++ b/fink_science/random_forest_snia/processor.py
@@ -474,7 +474,7 @@ def rfscore_rainbow_elasticc(
     >>> args += [F.col('diaObject.hostgal_snsep')]
     >>> args += [F.col('diaObject.hostgal_zphot')]
     >>> df = df.withColumn('pIa', rfscore_rainbow_elasticc(*args))
-    >>> print(df['pIa'].show())
+    >>> df.select('pIa').show()
     >>> df.filter(df['pIa'] > 0.5).count()
     80
     """