diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 14bd07f..0047db9 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -1,6 +1,7 @@
 image: jdrouet/docker-with-buildx:stable
 
 variables:
+  BUILD: 0
   DOCKER_HOST: tcp://docker:2375
   DOCKER_DRIVER: overlay2
   # Major.Minor.Release
@@ -19,5 +20,5 @@ build:
     - docker login -u $CI_REGISTRY_USER -p $CI_REGISTRY_PASSWORD $CI_REGISTRY
     - docker buildx create --use
     - docker buildx build --no-cache --platform linux/arm64,linux/amd64 -t $IMAGE_TAG --push .
-  only:
-    - main
+  rules:
+    - if: $CI_PIPELINE_SOURCE == "push" && ($CI_COMMIT_BRANCH == "main" || $CI_COMMIT_BRANCH == "staging") && $BUILD == "1"