diff --git a/AutomaticInterface/Tests/GeneratorTests.cs b/AutomaticInterface/Tests/GeneratorTests.cs
index 1e99c16..6d59195 100644
--- a/AutomaticInterface/Tests/GeneratorTests.cs
+++ b/AutomaticInterface/Tests/GeneratorTests.cs
@@ -1626,6 +1626,7 @@ class DemoClass
             // </auto-generated>
             //--------------------------------------------------------------------------------------------------
 
+            #nullable enable
             using System.CodeDom.Compiler;
             using AutomaticInterfaceAttribute;
             using System;
@@ -1643,6 +1644,7 @@ public partial interface IDemoClass
                     
                 }
             }
+            #nullable restore
 
             """;
         GenerateCode(code).Should().Be(expected);
@@ -1742,6 +1744,7 @@ class DemoClass
             // </auto-generated>
             //--------------------------------------------------------------------------------------------------
 
+            #nullable enable
             using System.CodeDom.Compiler;
             using AutomaticInterfaceAttribute;
             using System;
@@ -1759,6 +1762,7 @@ public partial interface IDemoClass
                     
                 }
             }
+            #nullable restore
 
             """;
         GenerateCode(code).Should().Be(expected);