Fixes belongsTo error