diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 9ddc781..a0de8fd 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -3,13 +3,13 @@ build:
     key: cache
     paths: [.cache]
   image: yourlabs/shlax
-  script: pip install -U --user -e . && CACHE_DIR=$(pwd)/.cache ./shlaxfile.py build
+  script: pip install -U --user -e .[cli] && CACHE_DIR=$(pwd)/.cache ./shlaxfile.py build
   stage: build
 
 test:
   image: yourlabs/python
   stage: build
-  script: pip install -U --user -e . && py.test -sv tests
+  script: pip install -U --user -e .[test] && py.test -sv tests
 
 pypi:
   image: yourlabs/python
diff --git a/setup.py b/setup.py
index b15babf..1579178 100644
--- a/setup.py
+++ b/setup.py
@@ -5,10 +5,9 @@ setup(
     name='shlax',
     versioning='dev',
     setup_requires='setupmeta',
-    install_requires=['cli2'],
     extras_require=dict(
-        full=[
-            'pyyaml',
+        cli=[
+            'cli2',
         ],
         test=[
             'pytest',