fixed model relationship issue