Fix class relation