Assign correctly