diff --git a/test/pytype_function.py b/test/pytype_function.py
index 2ca13c7f..7927629b 100755
--- a/test/pytype_function.py
+++ b/test/pytype_function.py
@@ -4,8 +4,8 @@
 """
 >>> from pytype_function_ext import *
 
->>> print (' ').join(func.__doc__.splitlines())
- func( (A)arg1) -> A :      C++ signature :         struct B func(struct B)
+>>> print func.__doc__.splitlines()[1]
+func( (A)arg1) -> A :
 
 """
 def run(args = None):