modify

Small change