diff --git a/scripts/superdwarf.rb b/scripts/superdwarf.rb
index e4e90434b..0592e83fd 100644
--- a/scripts/superdwarf.rb
+++ b/scripts/superdwarf.rb
@@ -3,13 +3,6 @@
 $superdwarf_onupdate ||= nil
 $superdwarf_ids ||= []
 
-def unregister
-    unless $superdwarf_onupdate.nil?
-        df.onupdate_unregister($superdwarf_onupdate)
-        $superdwarf_onupdate = nil
-    end
-end
-
 case $script_args[0]
 when 'add'
 	if u = df.unit_find
@@ -17,7 +10,8 @@ when 'add'
 
 		$superdwarf_onupdate ||= df.onupdate_register('superdwarf', 1) {
 			if $superdwarf_ids.empty?
-				unregister()
+				df.onupdate_unregister($superdwarf_onupdate)
+				$superdwarf_onupdate = nil
 			else
 				$superdwarf_ids.each { |id|
 					if u = df.unit_find(id) and not u.flags1.dead
@@ -32,11 +26,11 @@ when 'add'
 							when :Job2
 								a.data.job2.timer = 1
 							when :Attack
-                                # Attack execution timer; fires when reaches zero.
+								# Attack execution timer; fires when reaches zero.
 								a.data.attack.timer1 = 1
-                                # Attack completion timer: finishes action at zero.
-                                # An action must complete before target re-seleciton
-                                # occurs.
+								# Attack completion timer: finishes action at zero.
+								# An action must complete before target re-seleciton
+								# occurs.
 								a.data.attack.timer2 = 0
 							end
 						}
@@ -67,13 +61,8 @@ when 'del'
 		puts "Select a creature using 'v'"
 	end
 
-    if $superdwarf_ids.empty?
-        unregister()
-    end
-
 when 'clear'
 	$superdwarf_ids.clear
-    unregister()
 
 when 'list'
 	puts "current superdwarves:", $superdwarf_ids.map { |id| df.unit_find(id).name }