fixing up belongsTo