Fix for updating belongsTo