Edit command fix for #6