tracks/csharp/generators/Exercises/AllYourBase.cs in trackler-2.1.0.53 vs tracks/csharp/generators/Exercises/AllYourBase.cs in trackler-2.1.0.54

@@ -1,6 +1,7 @@
 using System;
+using Generators.Output;
 using Newtonsoft.Json.Linq;
 
 namespace Generators.Exercises
 {
     public class AllYourBase : Exercise
@@ -12,11 +13,19 @@
                 dynamic input = canonicalDataCase.Input;
                 input["input_digits"] = ((JArray)input["input_digits"]).Values<int>();
                 canonicalDataCase.Input = input;
 
                 canonicalDataCase.ExceptionThrown = canonicalDataCase.Expected is null ? typeof(ArgumentException) : null;
-                canonicalDataCase.UseInputParameters = true;
-                canonicalDataCase.UseExpectedParameter = canonicalDataCase.ExceptionThrown == null;
+                canonicalDataCase.UseVariablesForInput = true;
+                canonicalDataCase.UseVariableForExpected = canonicalDataCase.ExceptionThrown == null;
             }
+        }
+
+        protected override TestClass CreateTestClass()
+        {
+            var testClass = base.CreateTestClass();
+            testClass.UsingNamespaces.Add(typeof(ArgumentException).Namespace);
+
+            return testClass;
         }
     }
 }
\ No newline at end of file