lib/goaloc/generators/rails/model_test.rb.erb in mattknox-goaloc-0.3.0 vs lib/goaloc/generators/rails/model_test.rb.erb in mattknox-goaloc-0.4.0

@@ -1,8 +1,8 @@
 require File.dirname(__FILE__) + '/../test_helper'
 
-class <%= model.cs %>Test < ActiveSupport::TestCase
-  <% model.associations.each do |k, v| %>
+class <%= goal.cs %>Test < ActiveSupport::TestCase
+  <% goal.associations.each do |k, v| %>
     <% if :belongs_to == v[:type] %>
   <%= "should_belong_to :#{k}" %>
     <% end %>
   <% end %>
 end