def late_fee
days_late = Time.now.to_date - (due_date)
- return VideoPolicy.find_by_day(8).fee * days_late.to_i
+ if rentable.class == Video
+ return VideoPolicy.find_by_day(7).fee * days_late.to_i
+ else
+ return GamePolicy.find_by_day(7).fee * days_late.to_i
+ end
end
protected