Prettier and a minor bug fix