From e753aed127a99fd3be8f64e7b73260fa99b4f059 Mon Sep 17 00:00:00 2001
From: Rajan Mudambi <11376379+rmudambi@users.noreply.github.com>
Date: Tue, 24 Oct 2023 15:49:38 -0700
Subject: [PATCH] fix disease tests

---
 tests/disease/test_disease.py | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

diff --git a/tests/disease/test_disease.py b/tests/disease/test_disease.py
index f9c7a7102..dd2d04aba 100644
--- a/tests/disease/test_disease.py
+++ b/tests/disease/test_disease.py
@@ -1,7 +1,7 @@
 import numpy as np
 import pandas as pd
 import pytest
-from vivarium import InteractiveContext
+from vivarium import InteractiveContext, Component
 from vivarium.framework.state_machine import Transition
 from vivarium.framework.utilities import from_yearly
 from vivarium.testing_utilities import TestPopulation, build_table, metadata
@@ -358,10 +358,7 @@ def test_risk_deletion(base_config, base_plugins, disease):
 
     model = DiseaseModel(disease, initial_state=healthy, states=[healthy, sick])
 
-    class PafModifier:
-        @property
-        def name(self):
-            return "paf_modifier"
+    class PafModifier(Component):
 
         def setup(self, builder):
             builder.value.register_value_modifier(
@@ -370,7 +367,7 @@ def setup(self, builder):
                     build_table(paf, year_start, year_end),
                     key_columns=("sex",),
                     parameter_columns=["age", "year"],
-                    value_columns=None,
+                    value_columns=(),
                 ),
             )