Fix mass-assignment error