diff --git a/.travis.yml b/.travis.yml
index 1c646db8227..d86e882c566 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -66,13 +66,15 @@ matrix:
       env: TOXENV=py35
       script: &1
         # Additional dependencies
-        - pip3 install jsonschema==3.0.1
+        - pip3 install coverage coveralls jsonschema==3.0.1 
         # DSL tests
         - cd $TRAVIS_BUILD_DIR/sdk/python
-        - python3 setup.py install
-        - python3 tests/dsl/main.py
-        - python3 tests/compiler/main.py
-        - $TRAVIS_BUILD_DIR/sdk/python/tests/run_tests.sh
+        - python3 setup.py develop
+        - cd $TRAVIS_BUILD_DIR # Changing the current directory to the repo root for correct coverall paths
+        - coverage run --source=kfp --append sdk/python/tests/dsl/main.py
+        - coverage run --source=kfp --append sdk/python/tests/compiler/main.py
+        - coverage run --source=kfp --append -m unittest discover --verbose --start-dir sdk/python/tests --top-level-directory=sdk/python
+        - coveralls
 
         # Visualization test
         - cd $TRAVIS_BUILD_DIR/backend/src/apiserver/visualization
diff --git a/sdk/python/tests/components/test_components.py b/sdk/python/tests/components/test_components.py
index 047741e199a..2c621a000a5 100644
--- a/sdk/python/tests/components/test_components.py
+++ b/sdk/python/tests/components/test_components.py
@@ -17,7 +17,6 @@
 import unittest
 from pathlib import Path
 
-sys.path.insert(0, __file__ + '/../../../')
 
 import kfp
 import kfp.components as comp
diff --git a/sdk/python/tests/components/test_graph_components.py b/sdk/python/tests/components/test_graph_components.py
index dcee65d2096..b9d295546d2 100644
--- a/sdk/python/tests/components/test_graph_components.py
+++ b/sdk/python/tests/components/test_graph_components.py
@@ -17,7 +17,6 @@
 import unittest
 from pathlib import Path
 
-sys.path.insert(0, __file__ + '/../../../')
 
 import kfp.components as comp
 from kfp.components._structures import ComponentReference, ComponentSpec, ContainerSpec, GraphInputArgument, GraphSpec, InputSpec, InputValuePlaceholder, GraphImplementation, OutputPathPlaceholder, OutputSpec, TaskOutputArgument, TaskSpec