diff --git a/tests/test_dataset.py b/tests/test_dataset.py
index 9884f042..9aa73c86 100644
--- a/tests/test_dataset.py
+++ b/tests/test_dataset.py
@@ -791,9 +791,9 @@ def test_multicova_analysis_invalid_covariates(self):
         self.assertEqual(res.shape[1], 45)
 
     def test_get_protein_id_for_gene_name(self):
-        self.assertEqual(
+        with self.assertRaises(ValueError):
             self.obj._get_features_for_gene_name("MADE_UP_GENE"), "MADE_UP_GENE"
-        )
+
         self.assertEqual(
             self.obj._get_features_for_gene_name("ALDOC"),
             [