1 files changed, 16 insertions, 13 deletions
diff --git a/emlink.py b/emlink.py
index 7311f84a..f767ce58 100644
--- a/emlink.py
+++ b/emlink.py
@@ -10,21 +10,24 @@ import sys
 from tools import shared
 from tools.asm_module import AsmModule
 
-try:
-  me, main, side, out = sys.argv[:4]
-except:
-  print >> sys.stderr, 'usage: emlink.py [main module] [side module] [output name]'
-  sys.exit(1)
+def run():
+  try:
+    me, main, side, out = sys.argv[:4]
+  except:
+    print >> sys.stderr, 'usage: emlink.py [main module] [side module] [output name]'
+    sys.exit(1)
 
-print 'Main module:', main
-print 'Side module:', side
-print 'Output:', out
+  print 'Main module:', main
+  print 'Side module:', side
+  print 'Output:', out
 
-shared.try_delete(out)
+  shared.try_delete(out)
 
-main = AsmModule(main)
-side = AsmModule(side)
+  main = AsmModule(main)
+  side = AsmModule(side)
 
-side.relocate_into(main)
-main.write(out)
+  side.relocate_into(main)
+  main.write(out)
 
+if __name__ == '__main__':
+  run()