diff --git a/sdk/eventgrid/azure-eventgrid/tests/_mocks.py b/sdk/eventgrid/azure-eventgrid/tests/_mocks.py
index 7c979c4fd334..107d6d6175ed 100644
--- a/sdk/eventgrid/azure-eventgrid/tests/_mocks.py
+++ b/sdk/eventgrid/azure-eventgrid/tests/_mocks.py
@@ -1,5 +1,6 @@
 import json
 
+
 # storage cloud event
 cloud_storage_dict = {
     "id":"a0517898-9fa4-4e70-b4a3-afda1dd68672",
diff --git a/sdk/eventgrid/azure-eventgrid/tests/test_eg_consumer.py b/sdk/eventgrid/azure-eventgrid/tests/test_eg_consumer.py
index edf9450e59cf..755a7e15df4b 100644
--- a/sdk/eventgrid/azure-eventgrid/tests/test_eg_consumer.py
+++ b/sdk/eventgrid/azure-eventgrid/tests/test_eg_consumer.py
@@ -18,12 +18,15 @@
     cloud_storage_dict,
     cloud_storage_string,
     cloud_storage_bytes,
-    eg_storage_dict,
-    eg_storage_string,
-    eg_storage_bytes,
+    cloud_custom_dict,
+    cloud_custom_string,
+    cloud_custom_bytes,
     eg_custom_dict,
     eg_custom_string,
-    eg_custom_bytes
+    eg_custom_bytes,
+    eg_storage_dict,
+    eg_storage_string,
+    eg_storage_bytes
     )
 
 class EventGridConsumerTests(AzureMgmtTestCase):