diff --git a/src/backend/InvenTree/common/tests.py b/src/backend/InvenTree/common/tests.py
index 52a1eeca95..4cc16136bd 100644
--- a/src/backend/InvenTree/common/tests.py
+++ b/src/backend/InvenTree/common/tests.py
@@ -32,7 +32,7 @@ from InvenTree.unit_test import (
     PluginMixin,
     addUserPermission,
 )
-from part.models import Part, PartParameterTemplate
+from part.models import Part
 from plugin import registry
 
 from .api import WebhookView
@@ -45,6 +45,7 @@ from .models import (
     NotesImage,
     NotificationEntry,
     NotificationMessage,
+    ParameterTemplate,
     ProjectCode,
     SelectionList,
     SelectionListEntry,
@@ -2055,27 +2056,37 @@ class SelectionListTest(InvenTreeAPITestCase):
 
         # Add to parameter
         part = Part.objects.get(pk=1)
-        template = PartParameterTemplate.objects.create(
+        template = ParameterTemplate.objects.create(
             name='test_parameter', units='', selectionlist=self.list
         )
         rsp = self.get(
-            reverse('api-part-parameter-template-detail', kwargs={'pk': template.pk})
+            reverse('api-parameter-template-detail', kwargs={'pk': template.pk})
         )
         self.assertEqual(rsp.data['name'], 'test_parameter')
         self.assertEqual(rsp.data['choices'], '')
 
         # Add to part
-        url = reverse('api-part-parameter-list')
+        url = reverse('api-parameter-list')
         response = self.post(
             url,
-            {'part': part.pk, 'template': template.pk, 'data': 70},
+            {
+                'model_id': part.pk,
+                'model_type': 'part.part',
+                'template': template.pk,
+                'data': 70,
+            },
             expected_code=400,
         )
         self.assertIn('Invalid choice for parameter value', response.data['data'])
 
         response = self.post(
             url,
-            {'part': part.pk, 'template': template.pk, 'data': self.entry1.value},
+            {
+                'model_id': part.pk,
+                'model_type': 'part.part',
+                'template': template.pk,
+                'data': self.entry1.value,
+            },
             expected_code=201,
         )
         self.assertEqual(response.data['data'], self.entry1.value)