Fixed the belongsTo relation