1 files changed, 2 insertions, 1 deletions
diff --git a/add_seg b/add_seg
index c7fd521..e4fe22d 100755
--- a/add_seg
+++ b/add_seg
@@ -23,10 +23,11 @@ while line = STDIN.gets
   ext = '' if cfg[:nogz]
   s = "<seg"
   if cfg[:loo] then s += " exclude=\"#{i}\"" end
-  if cfg[:grammar] then s += " grammar=\"#{cfg[:grammar]}/grammar.#{i}#{ext}\"" end
   if index.size > 0
     puts s + " id=\"#{index[j]}\"> #{line.strip} </seg>"
+    if cfg[:grammar] then s += " grammar=\"#{cfg[:grammar]}/grammar.#{index[j]}#{ext}\"" end
   else
+    if cfg[:grammar] then s += " grammar=\"#{cfg[:grammar]}/grammar.#{i}#{ext}\"" end
     puts s + " id=\"#{i}\"> #{line.strip} </seg>"
   end
   i += 1