Update set-orientation.lua

DFHack · Jul 19, 2015 · f17f46d · f17f46d
1 parent 9f31929
commit f17f46d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/set-orientation.lua b/set-orientation.lua
@@ -11,6 +11,10 @@ local unit = nil
 if args.unit and tonumber(args.unit) then -- Check for unit declaration !REQUIRED
 	unit = df.unit.find(tonumber(args.unit))
 else
+	unit = dfhack.gui.getSelectedUnit()
+end
+
+if unit == nil then
 	print('No unit selected')
 	return
 end