diff --git a/tests/data/integration/expected_field.h b/tests/data/integration/expected_field.h
index a7ba085..49de737 100644
--- a/tests/data/integration/expected_field.h
+++ b/tests/data/integration/expected_field.h
@@ -119,13 +119,13 @@ struct DaisyField {
 
     // Display
     
-    daisy::OledDisplay<daisy::SSD130x4WireSpi128x64Driver>::Config display_config;
-    display_config.driver_config.transport_config.Defaults();
-    
-    display.Init(display_config);
-      display.Fill(0);
-      display.Update();
-     
+        daisy::OledDisplay<daisy::SSD130x4WireSpi128x64Driver>::Config display_config;
+        display_config.driver_config.transport_config.Defaults();
+        
+        display.Init(display_config);
+          display.Fill(0);
+          display.Update();
+         
 
     som.adc.Start();
   }
diff --git a/tests/data/integration/expected_patch.h b/tests/data/integration/expected_patch.h
index 7c9137b..b26f099 100644
--- a/tests/data/integration/expected_patch.h
+++ b/tests/data/integration/expected_patch.h
@@ -95,13 +95,13 @@ struct DaisyPatch {
 
     // Display
     
-    daisy::OledDisplay<daisy::SSD130x4WireSpi128x64Driver>::Config display_config;
-    display_config.driver_config.transport_config.Defaults();
-    
-    display.Init(display_config);
-      display.Fill(0);
-      display.Update();
-     
+        daisy::OledDisplay<daisy::SSD130x4WireSpi128x64Driver>::Config display_config;
+        display_config.driver_config.transport_config.Defaults();
+        
+        display.Init(display_config);
+          display.Fill(0);
+          display.Update();
+         
 
     // External Codec Initialization
     daisy::SaiHandle::Config sai_config[2];
diff --git a/tests/data/integration/expected_pod.h b/tests/data/integration/expected_pod.h
index de7ae94..cba8f7d 100644
--- a/tests/data/integration/expected_pod.h
+++ b/tests/data/integration/expected_pod.h
@@ -74,13 +74,13 @@ struct DaisyPod {
 
     // Display
     
-    daisy::OledDisplay<daisy::SSD130x4WireSpi128x64Driver>::Config display_config;
-    display_config.driver_config.transport_config.Defaults();
-    
-    display.Init(display_config);
-      display.Fill(0);
-      display.Update();
-     
+        daisy::OledDisplay<daisy::SSD130x4WireSpi128x64Driver>::Config display_config;
+        display_config.driver_config.transport_config.Defaults();
+        
+        display.Init(display_config);
+          display.Fill(0);
+          display.Update();
+         
 
     som.adc.Start();
   }
diff --git a/tests/integration_test.py b/tests/integration_test.py
index 384b4f9..17ce316 100644
--- a/tests/integration_test.py
+++ b/tests/integration_test.py
@@ -12,19 +12,6 @@
 
 
 class TestIntegration(unittest.TestCase):
-    __test__ = False
-
-    def test_custom(self):
-        self.maxDiff = None
-        header, info = json2daisy.generate_header_from_name('custom')
-        with open(path.join(data_path, 'integration', 'expected_custom.h'), 'r') as file:
-            self.assertEqual(header, file.read(), 'The output string should match "expected_custom.h" exactly')
-
-    def test_petal(self):
-        self.maxDiff = None
-        header, info = json2daisy.generate_header_from_name('petal')
-        with open(path.join(data_path, 'integration', 'expected_petal.h'), 'r') as file:
-            self.assertEqual(header, file.read(), 'The output string should match "expected_petal.h" exactly')
 
     def test_patch(self):
         self.maxDiff = None
@@ -38,11 +25,11 @@ def test_patch_init(self):
         with open(path.join(data_path, 'integration', 'expected_patch_init.h'), 'r') as file:
             self.assertEqual(header, file.read(), 'The output string should match "expected_patch_init.h" exactly')
 
-    def test_field(self):
+    def test_petal(self):
         self.maxDiff = None
-        header, info = json2daisy.generate_header_from_name('field')
-        with open(path.join(data_path, 'integration', 'expected_field.h'), 'r') as file:
-            self.assertEqual(header, file.read(), 'The output string should match "expected_field.h" exactly')
+        header, info = json2daisy.generate_header_from_name('petal')
+        with open(path.join(data_path, 'integration', 'expected_petal.h'), 'r') as file:
+            self.assertEqual(header, file.read(), 'The output string should match "expected_petal.h" exactly')
 
     def test_petal_125b_sm(self):
         self.maxDiff = None
@@ -50,14 +37,14 @@ def test_petal_125b_sm(self):
         with open(path.join(data_path, 'integration', 'expected_petal_125b_sm.h'), 'r') as file:
             self.assertEqual(header, file.read(), 'The output string should match "expected_petal_125b_sm.h" exactly')
 
-    def test_petal_125b_sm(self):
-        self.maxDiff = None
-        header, info = json2daisy.generate_header_from_name('petal_125b_sm')
-        with open(path.join(data_path, 'integration', 'expected_petal_125b_sm.h'), 'r') as file:
-          self.assertEqual(header, file.read(), 'The output string should match "expected_petal_125b_sm.h" exactly')
-
     def test_pod(self):
         self.maxDiff = None
         header, info = json2daisy.generate_header_from_name('pod')
         with open(path.join(data_path, 'integration', 'expected_pod.h'), 'r') as file:
             self.assertEqual(header, file.read(), 'The output string should match "expected_pod.h" exactly')
+
+    def test_field(self):
+        self.maxDiff = None
+        header, info = json2daisy.generate_header_from_name('field')
+        with open(path.join(data_path, 'integration', 'expected_field.h'), 'r') as file:
+            self.assertEqual(header, file.read(), 'The output string should match "expected_field.h" exactly')