Now we can use with rails 4