fix relationship issue