1 files changed, 7 insertions, 3 deletions
diff --git a/Software/Visual_Studio/Tango.UnitTesting/CodeGeneration_TST.cs b/Software/Visual_Studio/Tango.UnitTesting/CodeGeneration_TST.cs
index b4dda025b..6e546b306 100644
--- a/Software/Visual_Studio/Tango.UnitTesting/CodeGeneration_TST.cs
+++ b/Software/Visual_Studio/Tango.UnitTesting/CodeGeneration_TST.cs
@@ -1,7 +1,7 @@
 using System;
 using Microsoft.VisualStudio.TestTools.UnitTesting;
 using Tango.DAL.Remote.DB;
-using Tango.DAL.Remote.ObservableEntities;
+using Tango.DAL.Observables;
 using Tango.Settings;
 using System.Linq;
 using System.Collections;
@@ -26,7 +26,8 @@ namespace Tango.UnitTesting
             {
                 EntityCodeFile codeFile = new EntityCodeFile(Singularize(ObservableEntity<Object>.DalNameToStandardName(table.Name)))
                 {
-                    TableName = Singularize(table.Name)
+                    EntityName = Singularize(table.Name),
+                    TableName = table.Name,
                 };
 
                 foreach (var field in table.PropertyType.GenericTypeArguments.First().GetProperties().Skip(4))
@@ -75,8 +76,11 @@ namespace Tango.UnitTesting
                 foreach (var machine in machines)
                 {
                     machine.SerialNumber = "1111";
-                    machine.Save();
+                    machine.Organization.Address.AddressString = "HOW HOW HOW";
+                    machine.Save(db);
                 }
+
+                db.SaveChanges();
             }
         }