Fix bug with belongsTo