Fix problem for Active Record Association