Quick hack for rails 4 compatability

You probably don't actually want to merge this, but this makes it work for me with rails 4.  Thanks!