diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 84079cdeb17fb1e4ad7f51ddaaf8161d86f3ec1f..c989d9139677e5a4ec3691a3e9fc8b365416cde8 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -2,13 +2,13 @@
 include:
   - project: shore/ci-stuff
     file: templates/pre-commit.yml
-  - project: shore/ci-stuff
-    file: templates/python.yml
-
-variables:
-  SKIP_PYPI_UPLOAD: 1
 
-install wheel:
-  extends: .install-wheel
+test:
+  image: registry.shore.co.il/ci-images:python3
+  before_script:
+    - python3 -m pip install build twine
   script:
+    - python3 -m build
+    - twine check dist/*.whl
+    - python3 -m pip install dist/*.whl
     - sampleproject