diff --git a/.github/workflows/CI.yml b/.github/workflows/CI.yml
index 995e823..d1b946f 100644
--- a/.github/workflows/CI.yml
+++ b/.github/workflows/CI.yml
@@ -1,10 +1,15 @@
 name: CI
 
 on: 
-  # push:
-  # pull_request:
+  push:
+  pull_request:
   workflow_dispatch:
 
+# JUNGVI: Create a concurrency group per branch such that only the latest commit on a branch runs workflows
+concurrency:
+  group: ${{ github.workflow }}-${{ github.ref }}
+  cancel-in-progress: true
+
 jobs:
 
   build-deepquant:
diff --git a/pyproject.toml b/pyproject.toml
index 0534afe..81edbc1 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -18,10 +18,10 @@ authors = [
     {name = "Federico Brancasi", email = "fbrancasi@ethz.ch"},
 ]
 dependencies = [
-    "torch>=2.1.2",
+    "torch==2.1.2",
     "torchvision>=0.16.2",
     "torchaudio>=2.1.2",
-    "brevitas>=0.11.0",
+    "brevitas==0.11.0",
     "torchmetrics",
     "black",
     "isort",