Update set-orientation.lua

DFHack · Jul 19, 2015 · 35211d5 · 35211d5
1 parent f17f46d
commit 35211d5
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/set-orientation.lua b/set-orientation.lua
@@ -10,6 +10,10 @@ local unit = nil
 
 if args.unit and tonumber(args.unit) then -- Check for unit declaration !REQUIRED
 	unit = df.unit.find(tonumber(args.unit))
+	if unit == nil then
+		print('Bad unit ID')
+		return
+	end
 else
 	unit = dfhack.gui.getSelectedUnit()
 end
@@ -19,11 +23,6 @@ if unit == nil then
 	return
 end
 
-if unit == nil then
-	print('Bad unit ID')
-	return
-end
-
 if args.men and tonumber(args.men) then
 	men = tonumber(args.men)
 	if men < 0 or men > 2 then