diff --git a/InvenTree/build/models.py b/InvenTree/build/models.py
index 763686b765..b5f5aa10eb 100644
--- a/InvenTree/build/models.py
+++ b/InvenTree/build/models.py
@@ -17,7 +17,7 @@ from django.db.models import Sum
 from django.core.validators import MinValueValidator
 
 from stock.models import StockItem
-from part.models import BomItem
+from part.models import Part, BomItem
 
 
 class Build(models.Model):