diff --git a/python/altrios/tests/test_metric_calculator.py b/python/altrios/tests/test_metric_calculator.py
index 6fc754c5..e669950b 100644
--- a/python/altrios/tests/test_metric_calculator.py
+++ b/python/altrios/tests/test_metric_calculator.py
@@ -24,5 +24,5 @@ def test_dummy_sim(self):
             
         for info in scenario_infos:
             tkm = metric_calculator.calculate_freight_moved(info,'million tonne-mi')
-            self.assertEqual(tkm.filter(pl.col("Metric") == 'Mt-km').get_column("Units").len(), 1)
+            self.assertEqual(tkm.filter(pl.col("Metric") == pl.lit("Freight_Moved")).get_column("Units").len(), 1)