diff --git a/tests/src/test/scala/actionContainers/PythonActionContainerTests.scala b/tests/src/test/scala/actionContainers/PythonActionContainerTests.scala
index ce25ffcdd3bf552d0d9ce0449a5e3f9510b8aece..06d9e1efd2057d406af9a93a2e5110d0bca3a621 100644
--- a/tests/src/test/scala/actionContainers/PythonActionContainerTests.scala
+++ b/tests/src/test/scala/actionContainers/PythonActionContainerTests.scala
@@ -43,11 +43,12 @@ class PythonActionContainerTests extends BasicActionRunnerTests with WskActorSys
 
     testEcho(Seq {
         ("python", """
+          |from __future__ import print_function
           |import sys
-          |def main(dict):
-          |    print 'hello stdout'
-          |    print >> sys.stderr, 'hello stderr'
-          |    return dict
+          |def main(args):
+          |    print('hello stdout')
+          |    print('hello stderr', file=sys.stderr)
+          |    return args
           """.stripMargin)
     })