diff --git a/lib/propane/generator.rb b/lib/propane/generator.rb
index 839bd1b..539591f 100644
--- a/lib/propane/generator.rb
+++ b/lib/propane/generator.rb
@@ -191,7 +191,7 @@ class Propane
         end
         code = code.gsub(/\$(\d+)/) do |match|
           index = $1.to_i
-          "statevalues[$-1-n_states+#{index}].pvalue.v_#{rule.ptypename}"
+          "statevalues[$-1-n_states+#{index}].pvalue.v_#{rule.components[index - 1].ptypename}"
         end
       else
         code = code.gsub(/\$\$/) do |match|