diff --git a/data/core/commands/core.lua b/data/core/commands/core.lua
index e7bc3190..a3fd948c 100644
--- a/data/core/commands/core.lua
+++ b/data/core/commands/core.lua
@@ -107,12 +107,6 @@ command.add(nil, {
   ["core:open-user-module"] = function()
     local user_module_doc = core.open_doc(USERDIR .. "/init.lua")
     if not user_module_doc then return end
-    local doc_save = user_module_doc.save
-    user_module_doc.save = function(self)
-      doc_save(self)
-      core.reload_module("core.style")
-      core.load_user_directory()
-    end
     core.root_view:open_doc(user_module_doc)
   end,
 
diff --git a/data/core/commands/doc.lua b/data/core/commands/doc.lua
index e8eae510..3d1cd7b5 100644
--- a/data/core/commands/doc.lua
+++ b/data/core/commands/doc.lua
@@ -67,6 +67,10 @@ end
 
 local function save(filename)
   doc():save(filename)
+  if doc().filename == (USERDIR .. PATHSEP .. "init.lua") then
+    core.reload_module("core.style")
+    core.load_user_directory()
+  end
   core.log("Saved \"%s\"", doc().filename)
 end