diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index 5628812..da3e556 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -17,7 +17,7 @@ repos:
     rev: 7.1.1
     hooks:
       - id: flake8
-        args: [--max-line-length=128, '--exclude=./.*,build,dist,official*,helper_code.py,run_model.py,train_model.py,evaluate_model.py,remove_hidden_data.py,prepare_image_data.py,prepare_ptbxl_data.py,*.ipynb', '--ignore=E501,W503,E203,F841,E402,E231', --count, --statistics, --show-source]
+        args: [--max-line-length=128, '--exclude=./.*,build,dist,official*,torch_ecg,references,*.ipynb', '--ignore=E501,W503,E203,F841,E402,E231', --count, --statistics, --show-source]
   - repo: https://github.com/pycqa/isort
     rev: 5.13.2
     hooks: