Tweaks
Minor fix