diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 3795e1c..a17b63b 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -15,7 +15,12 @@ before_script:
   - virtualenv venv
   - source venv/bin/activate
 
-test-run:
+test:
+  script:
+    - pip install tox flake8
+    - tox -e py36,flake
+
+run:
   script:
   - pip install . jupyter
   - jupyter nbconvert --to markdown --execute README.ipynb
diff --git a/multi_bary_plot/__init__.py b/multi_bary_plot/__init__.py
index b762396..56ec5fa 100644
--- a/multi_bary_plot/__init__.py
+++ b/multi_bary_plot/__init__.py
@@ -1 +1,3 @@
 from .GenBary import GenBary
+
+__all__ = ['GenBary', ]
diff --git a/tox.ini b/tox.ini
index c201458..f41f7b5 100644
--- a/tox.ini
+++ b/tox.ini
@@ -1,8 +1,3 @@
-# tox (https://tox.readthedocs.io/) is a tool for running tests
-# in multiple virtualenvs. This configuration file will run the
-# test suite on all supported python versions. To use it, "pip install tox"
-# and then run "tox" from this directory.
-
 [tox]
 envlist = flake8