Minor fix

Just some style changes, nothing major.