diff --git a/.github/workflows/test_full.yml b/.github/workflows/test_full.yml
index 91da848..60b90d9 100644
--- a/.github/workflows/test_full.yml
+++ b/.github/workflows/test_full.yml
@@ -1,4 +1,4 @@
-name: Tests
+name: Tests Full
 
 on:
   schedule:
diff --git a/.github/workflows/test_pr.yml b/.github/workflows/test_pr.yml
index c88ae6b..dde9ca9 100644
--- a/.github/workflows/test_pr.yml
+++ b/.github/workflows/test_pr.yml
@@ -1,4 +1,4 @@
-name: Tests
+name: Tests Fast
 
 on:
   push:
diff --git a/README.md b/README.md
index 5db4f8f..ac04104 100644
--- a/README.md
+++ b/README.md
@@ -4,6 +4,7 @@
 
 [![Test In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/drive/1zGm4VeXsJ-0x6A5_icnknE7mbJ0knUig?usp=sharing)
 [![Tests PR](https://github.com/vanderschaarlab/hyperimpute/actions/workflows/test_pr.yml/badge.svg)](https://github.com/vanderschaarlab/hyperimpute/actions/workflows/test_pr.yml)
+[![Tests Full](https://github.com/vanderschaarlab/hyperimpute/actions/workflows/test_full.yml/badge.svg)](https://github.com/vanderschaarlab/hyperimpute/actions/workflows/test_full.yml)
 [![Tutorials](https://github.com/vanderschaarlab/hyperimpute/actions/workflows/test_tutorials.yml/badge.svg)](https://github.com/vanderschaarlab/hyperimpute/actions/workflows/test_tutorials.yml)
 [![Documentation Status](https://readthedocs.org/projects/hyperimpute/badge/?version=latest)](https://hyperimpute.readthedocs.io/en/latest/?badge=latest)
 
diff --git a/src/hyperimpute/plugins/imputers/_hyperimpute_internals.py b/src/hyperimpute/plugins/imputers/_hyperimpute_internals.py
index ffd21f1..f6ae61a 100644
--- a/src/hyperimpute/plugins/imputers/_hyperimpute_internals.py
+++ b/src/hyperimpute/plugins/imputers/_hyperimpute_internals.py
@@ -695,7 +695,7 @@ def _setup(self, X: pd.DataFrame) -> pd.DataFrame:
                 )
 
             if X[col].dtype == "object" or self._is_categorical(X, col):
-                # TODOD: One hot encoding for smaller cnt
+                # TODO: One hot encoding for smaller size features
 
                 existing_vals = X[col][X[col].notnull()]