diff --git a/src/backend/InvenTree/part/test_migrations.py b/src/backend/InvenTree/part/test_migrations.py
index 4f4c726788..10d528e2a2 100644
--- a/src/backend/InvenTree/part/test_migrations.py
+++ b/src/backend/InvenTree/part/test_migrations.py
@@ -356,4 +356,4 @@ class TestPartParameterDeletion(MigratorTestCase):
             self.assertEqual(len(params), 3)
 
             for unit in self.UNITS:
-                self.assertTrue(params.filter(units=unit).exists())
+                self.assertTrue(params.filter(template__units=unit).exists())