The quick release/keyless chuck on my Milwaukee is going — bits smaller than 1/4" slip.

Obviously I can get the replacement part from Milwaukee but is it worth upgrading to another chuck e.g. Röhm? '

Or is there something I can adjust to make this thing work better?