Fixes mass-assign error