diff --git a/InvenTree/build/test_build.py b/InvenTree/build/test_build.py
index b572feb14b..30fe8c488b 100644
--- a/InvenTree/build/test_build.py
+++ b/InvenTree/build/test_build.py
@@ -8,8 +8,9 @@ from django.db.utils import IntegrityError
 from InvenTree import status_codes as status
 
 from build.models import Build, BuildItem, get_next_build_number
-from stock.models import StockItem
 from part.models import Part, BomItem
+from stock.models import StockItem
+from stock.tasks import delete_old_stock_items
 
 
 class BuildTest(TestCase):
@@ -352,6 +353,11 @@ class BuildTest(TestCase):
         # the original BuildItem objects should have been deleted!
         self.assertEqual(BuildItem.objects.count(), 0)
 
+        self.assertEqual(StockItem.objects.count(), 8)
+
+        # Clean up old stock items
+        delete_old_stock_items()
+
         # New stock items should have been created!
         self.assertEqual(StockItem.objects.count(), 7)
 
diff --git a/InvenTree/stock/tests.py b/InvenTree/stock/tests.py
index 52f934544f..cb1bd406bb 100644
--- a/InvenTree/stock/tests.py
+++ b/InvenTree/stock/tests.py
@@ -332,7 +332,7 @@ class StockTest(TestCase):
         w1 = StockItem.objects.get(pk=100)
         w2 = StockItem.objects.get(pk=101)
 
-        self.assertFalse(w2.scheduled_for_depletion)
+        self.assertFalse(w2.scheduled_for_deletion)
 
         # Take 25 units from w1 (there are only 10 in stock)
         w1.take_stock(30, None, notes='Took 30')
@@ -346,7 +346,7 @@ class StockTest(TestCase):
 
         # w2 should now be marked for future deletion
         w2 = StockItem.objects.get(pk=101)
-        self.assertTrue(w2.scheduled_for_depletion)
+        self.assertTrue(w2.scheduled_for_deletion)
 
         from stock.tasks import delete_old_stock_items