X-Git-Url: https://projects.mako.cc/source/selectricity/blobdiff_plain/adfcf4dd7d989d0560e2a8f6a590dd10c5a49a08..c4072579c6b3e80d8f05768a9a68bc9ad7490b3c:/app/models/election.rb

diff --git a/app/models/election.rb b/app/models/election.rb
index 08ef1a5..7c15dcf 100644
--- a/app/models/election.rb
+++ b/app/models/election.rb
@@ -86,7 +86,7 @@ class Election < ActiveRecord::Base
   end
   
   def quickvote?
-    self.class == 'QuickVote'
+    self.class == QuickVote
   end
 
   def active?
@@ -141,11 +141,11 @@ class Election < ActiveRecord::Base
       # skip if the voter has not voted or has an unconfirmed vote
       next unless voter.voted?
 
-      plurality_tally << voter.vote.rankings.sort[0].candidate.id
+      plurality_tally << voter.vote.rankings.sort[0].candidate_id
       approval_tally << voter.vote.rankings.sort[0..1].collect \
-        { |ranking| ranking.candidate.id }
+        { |ranking| ranking.candidate_id }
       preference_tally << voter.vote.rankings.sort.collect \
-        { |ranking| ranking.candidate.id }
+        { |ranking| ranking.candidate_id }
     end
     
     @plurality_result = PluralityVote.new(plurality_tally).result