Allow for activerecord greater than 4.0