From bf4ff79eacea6ba008ad3ac72003874906d18d26 Mon Sep 17 00:00:00 2001
From: SnpM <jpthek9@gmail.com>
Date: Tue, 16 Aug 2016 21:13:12 -0600
Subject: [PATCH] Fix environment saving

---
 Integration/Environment/DefaultSaver.cs                   | 3 +++
 Integration/Environment/Editor/EditorEnvironmentHelper.cs | 5 ++---
 Integration/Environment/EnvironmentHelper.cs              | 1 +
 Integration/Environment/EnvironmentSaver.cs               | 3 ++-
 4 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/Integration/Environment/DefaultSaver.cs b/Integration/Environment/DefaultSaver.cs
index 3c37dbc7..c632d5ad 100755
--- a/Integration/Environment/DefaultSaver.cs
+++ b/Integration/Environment/DefaultSaver.cs
@@ -65,6 +65,9 @@ void SaveObjects () {
 			for (int i = 0; i < _environmentObjects.Length; i++)
 			{
 				_environmentObjects[i].Save();
+#if UNITY_EDITOR
+				UnityEditor.EditorUtility.SetDirty(_environmentObjects[i]);
+#endif
 			}
         }
         static bool IsAgent (object obj) {
diff --git a/Integration/Environment/Editor/EditorEnvironmentHelper.cs b/Integration/Environment/Editor/EditorEnvironmentHelper.cs
index b10ac288..b5fd5f34 100755
--- a/Integration/Environment/Editor/EditorEnvironmentHelper.cs
+++ b/Integration/Environment/Editor/EditorEnvironmentHelper.cs
@@ -20,9 +20,8 @@ public override void OnInspectorGUI()
             if (GUILayout.Button("Scan and Save")) {
                 
                 saver.ScanAndSave();
-                EditorUtility.SetDirty(target);
-                serializedObject.Update();
-            }
+
+			}
 
         }
 
diff --git a/Integration/Environment/EnvironmentHelper.cs b/Integration/Environment/EnvironmentHelper.cs
index 945d0f13..6ae1e9a5 100755
--- a/Integration/Environment/EnvironmentHelper.cs
+++ b/Integration/Environment/EnvironmentHelper.cs
@@ -27,6 +27,7 @@ public void ScanAndSave () {
             InitializeEnvironmentFromObject();
             foreach (EnvironmentSaver saver in Savers) {
                 saver.Save();
+				UnityEditor.EditorUtility.SetDirty(saver);
             }
         }
 
diff --git a/Integration/Environment/EnvironmentSaver.cs b/Integration/Environment/EnvironmentSaver.cs
index 57a114c7..78344d51 100755
--- a/Integration/Environment/EnvironmentSaver.cs
+++ b/Integration/Environment/EnvironmentSaver.cs
@@ -7,7 +7,8 @@ public abstract class EnvironmentSaver : MonoBehaviour
     {
         public void Save() {
             OnSave ();
-        }
+
+		}
         protected virtual void OnSave () {
 
         }