diff --git a/.travis.yml b/.travis.yml
index 5cf9696..8db1ade 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -11,7 +11,7 @@ env:
   - TEST_CMD="python pyunit.py" SCRIPT_OPT="--framework=pyunit" FRAMEWORK=PyUnit
 
 script:
-  - $TEST_CMD | python annotate.py --tool python --tee --name "$FRAMEWORK log"
+  - $TEST_CMD 2>&1 | python annotate.py --tool python --tee --name "$FRAMEWORK log"
 
 after_script:
   - curl -s https://raw.githubusercontent.com/report-ci/scripts/master/upload.py | python - $SCRIPT_OPT
\ No newline at end of file
diff --git a/pytest_example.py b/pytest_example.py
index 5e9e835..e3a090d 100644
--- a/pytest_example.py
+++ b/pytest_example.py
@@ -1,11 +1,12 @@
 
 # content of test_sample.py
 def inc(x):
+def inc(x,y):
     return x + 1
 
 
-def foo():
-    assert True
+def foo(x):
+    assert x
 
 def test_foo(): foo()
 
diff --git a/pyunit.py b/pyunit.py
index 7ac7e7f..9e365af 100644
--- a/pyunit.py
+++ b/pyunit.py
@@ -1,5 +1,4 @@
 import unittest
-import xmlrunner
 
 class SimpleTestCase(unittest.TestCase):