diff --git a/.github/workflows/run_tests.yml b/.github/workflows/run_tests.yml
index f1dad637..031a75f6 100644
--- a/.github/workflows/run_tests.yml
+++ b/.github/workflows/run_tests.yml
@@ -27,6 +27,7 @@ jobs:
         run: |
           python -m pip install --upgrade pip
           if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
+          pip install torch-geometric>=2.5.2
       - name: Test with pytest
         run: |
           pytest  -v -s
diff --git a/README.md b/README.md
index bd0d901f..1bdc6602 100644
--- a/README.md
+++ b/README.md
@@ -1,10 +1,5 @@
-![Linting](https://github.com/SimonKamuk/neural-lam/actions/workflows/pre-commit.yml/badge.svg?branch=feature_add_tests)
-![Automatic tests](https://github.com/SimonKamuk/neural-lam/actions/workflows/run_tests.yml/badge.svg?branch=feature_add_tests)
-
-<!---
 ![Linting](https://github.com/mllam/neural-lam/actions/workflows/pre-commit.yml/badge.svg)
 ![Automatic tests](https://github.com/mllam/neural-lam/actions/workflows/run_tests.yml/badge.svg)
--->
 
 <p align="middle">
     <img src="figures/neural_lam_header.png" width="700">
diff --git a/requirements.txt b/requirements.txt
index 16a29a18..9309eea4 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -10,7 +10,6 @@ Cartopy>=0.22.0
 pyproj>=3.4.1
 tueplots>=0.0.8
 plotly>=5.15.0
-torch-geometric>=2.5.2
 
 # for dev
 pre-commit>=2.15.0