diff --git a/mods/lzr_levels/init.lua b/mods/lzr_levels/init.lua
index cbf4230..34cc717 100644
--- a/mods/lzr_levels/init.lua
+++ b/mods/lzr_levels/init.lua
@@ -221,9 +221,12 @@ minetest.register_chatcommand("restart", {
 	params = "",
 	description = S("Restart current level"),
 	func = function(name, param)
-		if lzr_gamestate.get_state() == lzr_gamestate.LEVEL then
+		local state = lzr_gamestate.get_state()
+		if state == lzr_gamestate.LEVEL or state == lzr_gamestate.EDITOR then
 			lzr_levels.start_level(current_level)
 			return true
+		elseif state == lzr_gamestate.LEVEL_COMPLETE then
+			return false, S("Can’t restart level right now.")
 		else
 			return false, S("Not playing in a level!")
 		end