diff --git a/streaming/src/main/scala/org/apache/spark/streaming/api/python/PythonDStream.scala b/streaming/src/main/scala/org/apache/spark/streaming/api/python/PythonDStream.scala
index 59bb2ed5fa042..5ab15f717903e 100644
--- a/streaming/src/main/scala/org/apache/spark/streaming/api/python/PythonDStream.scala
+++ b/streaming/src/main/scala/org/apache/spark/streaming/api/python/PythonDStream.scala
@@ -174,7 +174,7 @@ class PythonTransformedDStream (parent: DStream[_], @transient pfunc: PythonTran
         // try to use the result as a template
         r.get match {
           case pyrdd: PythonRDD =>
-            if (pyrdd.parent(0) == rdd) {
+            if (pyrdd.firstParent == rdd) {
               // only one PythonRDD
               lastResult = pyrdd
             } else {