diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 10123f3..b8fb8b4 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -11,5 +11,5 @@ pypi:
   only: [tags]
   script: pypi-release
   stage: deploy
-test: {image: 'yourlabs/shlax:$CI_COMMIT_SHORT_SHA', script: pip install -U --user
-    -e . && ./shlaxfile.py -d test, stage: test}
+test: {image: yourlabs/python, script: 'pip install -U --user -e .[test] && py.test
+    -svv tests', stage: build}
diff --git a/shlaxfile.py b/shlaxfile.py
index 685cd93..3921ed5 100755
--- a/shlaxfile.py
+++ b/shlaxfile.py
@@ -41,7 +41,7 @@ gitlabci = GitLabCI(
         image='yourlabs/python',
     ),
     build=dict(
-        stage='test',
+        stage='build',
         image='yourlabs/shlax',
         script='pip install -U --user -e . && CACHE_DIR=$(pwd)/.cache ./shlaxfile.py -d shlax build push',
         cache=dict(paths=['.cache'], key='cache'),