Bug/typo fixes