diff --git a/Examples/test-suite/python_pythoncode.i b/Examples/test-suite/python_pythoncode.i
index d3593984f..70474d44c 100644
--- a/Examples/test-suite/python_pythoncode.i
+++ b/Examples/test-suite/python_pythoncode.i
@@ -5,17 +5,19 @@
 
 struct TYPE {
 %pythoncode %{
-   def one():
+    def one():
+        a = 1
 # Comment XXXX
-      return 1
+        return a
 %}
 };
 
 %define %bar
 %pythoncode %{
-   def one():
-      # Comment XXXX
-      return 1
+    def one():
+        a = 1
+        #
+        return a
 %}
 %enddef