Update try to work with rails 4.0