diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 1b11106..f96fbc8 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -27,12 +27,10 @@ stages:
 
 docker-build:
   extends: .docker-build
-  script:
-    - !reference [.docker-kaniko-build-config, script]
-    - !reference [.docker-kaniko-build-script, script]
   needs:
     - job: dvc-pull
       artifacts: true
+    - !reference [.needs-versioning, needs] # leave this line as is
   
 ###################
 # INTEGRATION TESTS
diff --git a/pyproject.toml b/pyproject.toml
index ef78333..3e698cb 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "image-classification-service"
-version = "2.4.1"
+version = "2.6.1"
 description = ""
 authors = ["Team Research <research@knecon.com>"]
 readme = "README.md"