Just a couple of tiny changes