fix belongs_to problem;