1 files changed, 7 insertions, 5 deletions
diff --git a/src/lua/common/je_lua_object.h b/src/lua/common/je_lua_object.h
index a81f57c..6e86508 100644
--- a/src/lua/common/je_lua_object.h
+++ b/src/lua/common/je_lua_object.h
@@ -31,19 +31,19 @@ namespace JinEngine
                 return shared->getObject<T>();
             }
 
-            void setDependency(uint key, Shared* shared);
+            void setDependency(uint key, LuaObject* dep);
 
             void removeDependency(uint key);
 
-            void removeDependency(Shared* dep);
+            void removeDependency(LuaObject* dep);
 
             bool isDependOn(uint key);
 
-            bool isDependOn(Shared* shared);
+            bool isDependOn(LuaObject* shared);
 
             void clearDependencies();
 
-            Shared* getDependency(uint key);
+            LuaObject* getDependency(uint key);
 
             int getDependenciesCount();
 
@@ -51,12 +51,14 @@ namespace JinEngine
             // Lua state object.
             //////////////////////////////////////////////////////////////////////////////////////////////////////
 
-            using DepsMap = std::map<uint, Shared*>;
+            using DepsMap = std::map<uint, LuaObject*>;
 
             lua_State* state;
 
             Shared* shared;
 
+            const char* type;
+
             DepsMap* dependencies;
 
         };