Make compatible with rails 4