diff --git a/datacenterlight/tests.py b/datacenterlight/tests.py
index 9ec2df96..dcbb727f 100644
--- a/datacenterlight/tests.py
+++ b/datacenterlight/tests.py
@@ -101,7 +101,7 @@ class CeleryTaskTestCase(TestCase):
         cpu = specs.get('cpu')
         memory = specs.get('memory')
         disk_size = specs.get('disk_size')
-        VMPricing.objects.create(
+        vm_pricing = VMPricing.objects.create(
             name="default",
             cores_unit_price=5,
             ram_unit_price=2,
@@ -146,7 +146,8 @@ class CeleryTaskTestCase(TestCase):
             price=specs['price'],
             vm_id=0,
             customer=stripe_customer,
-            billing_address=billing_address
+            billing_address=billing_address,
+            vm_pricing=vm_pricing
         )
 
         async_task = create_vm_task.delay(