Fixed error on model.rb