diff --git a/wani.py b/wani.py
index 64ce126..43f50f6 100644
--- a/wani.py
+++ b/wani.py
@@ -1360,12 +1360,12 @@ def main(structures_file, args):
     word_renderer.generate_renders()
     colocation_ids.determine_colocation_dispersions()
 
+    # figure out representations!
     if args.output:
-        # figure out representations!
         colocation_ids.set_representations(structures, word_renderer)
-        Writer.make_output_writer(args).write_out(structures, colocation_ids)
-    if args.all:
-        Writer.make_all_writer(args).write_out(structures, colocation_ids)
+
+    Writer.make_output_writer(args, colocation_ids).write_out(structures, colocation_ids)
+    Writer.make_all_writer(args, colocation_ids).write_out(structures, colocation_ids)
 
 if __name__ == '__main__':
     parser = argparse.ArgumentParser(