diff --git a/.github/workflows/run_tests.yml b/.github/workflows/run_tests.yml
index 1a21b8c1..1ebc3480 100644
--- a/.github/workflows/run_tests.yml
+++ b/.github/workflows/run_tests.yml
@@ -129,6 +129,6 @@ jobs:
         run: nox --force-color --session=coverage -- xml -i
 
       - name: Upload coverage report
-        uses: codecov/codecov-action@v5.1.1
+        uses: codecov/codecov-action@v5.3.1
         with:
           token: ${{ secrets.CODECOV_TOKEN }}
diff --git a/drevalpy/models/drp_model.py b/drevalpy/models/drp_model.py
index fb6b34f6..26a6d37b 100644
--- a/drevalpy/models/drp_model.py
+++ b/drevalpy/models/drp_model.py
@@ -213,6 +213,15 @@ def get_concatenated_features(
             cell_line_input=cell_line_input,
             drug_input=drug_input,
         )
+        if drug_view is not None:
+            if drug_view not in inputs:
+                raise ValueError(f"Expected drug_view '{drug_view}' to be in inputs, but it was not. Inputs: {inputs}")
+        if cell_line_view is not None:
+            if cell_line_view not in inputs:
+                raise ValueError(
+                    f"Expected cell_line_view '{cell_line_view}' to be in inputs, but it was not. Inputs: {inputs}"
+                )
+
         cell_line_features = None if cell_line_view is None else inputs.get(cell_line_view)
         drug_features = None if drug_view is None else inputs.get(drug_view)