From 5927bbfc3e50f24faa7f2256e174531ce17bb8a7 Mon Sep 17 00:00:00 2001
From: Matthias Mair <code@mjmair.com>
Date: Wed, 26 Nov 2025 02:15:10 +0100
Subject: [PATCH] fix test

---
 src/backend/InvenTree/part/test_category.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/backend/InvenTree/part/test_category.py b/src/backend/InvenTree/part/test_category.py
index c7672245e5..85d6d6c246 100644
--- a/src/backend/InvenTree/part/test_category.py
+++ b/src/backend/InvenTree/part/test_category.py
@@ -3,9 +3,9 @@
 from django.core.exceptions import ValidationError
 from django.test import TestCase
 
-from common.models import InvenTreeSetting
+from common.models import InvenTreeSetting, Parameter, ParameterTemplate
 
-from .models import Part, PartCategory, PartParameter, PartParameterTemplate
+from .models import Part, PartCategory
 
 
 class CategoryTest(TestCase):
@@ -164,8 +164,8 @@ class CategoryTest(TestCase):
             for fastener in fasteners:
                 self.assertIsInstance(fastener, Part)
                 for parameter in fastener.parameters.all():
-                    self.assertIsInstance(parameter, PartParameter)
-                    self.assertIsInstance(parameter.template, PartParameterTemplate)
+                    self.assertIsInstance(parameter, Parameter)
+                    self.assertIsInstance(parameter.template, ParameterTemplate)
 
             # Test number of unique parameters
             self.assertEqual(