Allow use with rails 4